牛客网刷题--剑指offer(两个栈实现队列)

牛客网给出的题目要求是用两个栈数据结构实现一个队列的push和pop操作。
两个栈实现队列这是很好理解的。因为栈的特性是后进先出(LIFO)，而队列的特性是先进先出(FIFO)，那两个栈连在一块儿，第一个栈先进后出，然后从第一个栈再入第二个栈，还是先进后出，这样，先是倒序入第一栈，然后再倒序入第二个栈，倒序的倒序就是正序，所以若将两个栈看成一个整体，还是正序入，正序出，就是队列的特性：先进先出。
既然两个栈能实现一个队列，那应该怎么实现呢？
假设有一串正序数列1，2，3，4，5。
第一次用队：将1入队
第二次用队：将1出队
第三次用队：将2，3，4入队
第四次用队：将2，3出队
第五次用队：将5入队
第六次用队：将5出队
经过这六次模拟用队，我们能够发现：
1.入队一定是将元素压入到第一个栈中。
2.出队时候需要做判断：如果第二个栈不为空，那直接出栈（即出队）；若第二个栈为空，那就必须得把第一个栈中的所有元素都入第二个栈后再出栈，这样就保证了序列有序。
根据这个逻辑：写出的代码。
牛客网规定的这个pop函数是能够返回一个值的，可是根据C++习惯，pop是不会返回值的。而我又是根据C++习惯，没有写返回值，所以第一次编译没有被通过，修改了之后。编译上传通过了所有的测试用例。
现将代码粘贴如下，以供后面查阅：

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

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

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