剑指offer 用两个栈实现队列

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

代码：

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

我的笔记：由于栈的特点，先进后出，而队列的特点是先进先出，因此在入栈 push 时无异，但出栈 pop 时需要两个栈来交替操作，即收到 pop 指令时，先检测第二栈 stack2 是否为空，若不为空，则直接出栈栈顶元素；若为空，则将 stack1 中元素迭代 push 到 stack2 中，直至 stack1 清空。