Java教程

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

本文主要是介绍算法题--用两个栈实现队列,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

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;	//出
};

拓展

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

模拟过程

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


这篇关于算法题--用两个栈实现队列的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!