【剑指offer】栈与队列——005 用两个栈实现队列

---

---

题目内容

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

题目分析

众所周知，队列是“先进先出”思想，栈是“先进后出”思想，题目的要求就是我们要让最先入队的元素最先出队。很容易就可以想到，把第一个栈里面的元素逐个保存到第二个栈中，这样原来的栈底，就变成了栈顶，达到了出栈要求。

解题

解法一

以stack1为主栈，stack2为辅助栈。主要思路如下：

• 入队时直接通过stack1的push()操作完成；
• 出队时，先把stack1里的元素全部出栈并压栈到stack2中，返回stack2的栈顶元素后再把所有元素压回stack1中。

代码

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }
 
    int pop() {
        while(!stack1.empty()){
            stack2.push(stack1.top());
            stack1.pop();
        }
        int value = stack2.top();
        stack2.pop();
        while(!stack2.empty()){
            stack1.push(stack2.top());
            stack2.pop();
        }
        return value;
    }
 
private:
    stack<int> stack1;
    stack<int> stack2;
};

优化解法

后来看了大神们的分析，发现自己的方法时间上有些复杂。过程如下：

操作	stack1	stack2
初始	{a, b, c}	{}
出队时	{}	{c, b, a}
出队后	{}	{c, b}
入队d	{d}	{c, b}
出队	{d}	{c}

通过以上过程可以发现，stack2不为空时，stack1的入队，stack2的出队是互不影响的，即过程简化为：

• 入队时直接通过stack1的push()操作完成；
• 出队时，若stack2为空，先把stack1里的元素全部出栈并压栈到stack2中，记录stack2的栈顶元素后执行pop()操作；否则，直接记录stack2的栈顶元素后执行pop()操作。

代码

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

    int pop() {
        int value = 0;
        if(!stack2.empty()){
            value = stack2.top();
            stack2.pop();
        }else{
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
            value = stack2.top();
            stack2.pop();
        }
        return value;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};

以上思路为个人想法和网络各位大佬的题解的结合，欢迎讨论与指正。