5. 用两个栈实现队列

题目

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

思想

• 队列是先进先出，栈是先进后出。
• 两个栈，stack1用来push，stack2用来pop.
• push的时候直接把元素入stack1。
• pop的时候要考虑一下，因为要把先入stack1的元素pop出，所以将stack1的元素依次出栈，并且放入stack2中，这样stack2中元素的出栈顺序就相当于模拟的队列出队列的顺序。
• stack2不为空的话，不能接收stack1，会导致元素顺序的混乱。

代码

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

    int pop() {
        int res = -1;
        
        if ( stack2.empty() ) {
            while ( !stack1.empty() ) {
                stack2.push( stack1.top() );
                stack1.pop();
            }
        }
         
        if ( stack2.empty() ) return res; // 如果队列为空，返回-1表示异常
        
        res = stack2.top();
        stack2.pop();
        
        return res;
    }
private:
    stack<int> stack1;
    stack<int> stack2;
};