算法题--用两个栈实现队列

7

要求

时间限制：1秒 空间限制：32768K

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型

解题思路

利用栈可以颠倒一个序列的顺序这个特性来思考

像这类题目只要模拟一下就能找到答案：先加入一些元素，然后思考如何弹出你想要的元素，这样就解决了弹出的问题；再思考添加的问题

这道题中，stack1用于入队，stack2用于出队，只是出队是要注意：要保证stack2不为空时才可以出队；如果它为空，就要先将stack1中所有元素弹出到stack2中，再从stack2中弹出一个元素；如果它不为空，直接从stack2中弹出一个元素即可

代码

class Solution
{
public:
  void push(int node)
  {
    stack1.push(node);
  }

  int pop()
  {
    if(stack2.empty())
    {
      while(!stack1.empty())
      {
         stack2.push(stack1.top());
         stack1.pop();
      }
    }
      
    if(stack2.empty())
    {
      cout << "队列为空" << endl;
      abort();
    }
    
    int res = stack2.top();
    stack2.pop();
    return res;
  }

private:
    stack<int> stack1;    //入
    stack<int> stack2;    //出
};

拓展

用两个队列来模拟栈，思考方式和这道题一样

两个队列的元素要不停的倒来倒去

---

小弟才浅，如果本篇文章有任何错误和建议，欢迎大家留言

感谢大家的点赞、收藏

微信搜「三年游戏人」第一时间阅读最新内容，获取一份收集多年的书籍包 以及 优质工作内推