剑指Offer——用两个栈实现队列

题目描述

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

解题思路

队列是先进先出，而栈是先进后出，想要做到队列的效果，可以先将数值压入stack1，然后一次从stack1中弹出，压入到stack2中，队列想要pop取值时，从stack2中pop，则符合先进先出的规则。

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

以上为第一版本，但每次pop时，都要将数值从stack1导入stack2，取出栈顶元素后，又从stack2导回stack1，其实是没有必要的，将数值从stack1导入stack2后，不需要导回，下次需要pop时，如果stack2不为空，直接从stack2中pop即可，当stack2为空，stack1不为空时，再从stack1导入stack2。修改后代码如下：

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