剑指offer：用两个栈实现队列（栈和队列）

题目描述

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

分析：

输入1，2，3入队，则输出也为1，2，3。由于我们只有栈，考虑全部压入第一个栈，则有

stack 1

    3

    2

    1

这时POP应该先输出1，考虑到用第二个栈储存第一个栈POP出来的元素，恰好是队列的输出顺序

stack1                 stack2

                                1

                                2

                                3

可以得出结论，push的时候，压入stack1，pop的时候，如果stack2有元素，则直接pop出stack2的栈顶元素，否则，把stack1的元素压入stack2，如何再输出stack2的栈顶元素。

代码：

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