剑指Offer 用两个栈实现队列

本题借鉴了博客https://blog.csdn.net/malele4th/article/details/79316033

先说明C++中栈和队列的一些基本操作。

栈的基本操作：

s.push(item);       //将item压入栈顶  
s.pop();            //删除栈顶的元素，但不会返回  
s.top();            //返回栈顶的元素，但不会删除  
s.size();           //返回栈中元素的个数  
s.empty();          //检查栈是否为空，如果为空返回true，否则返回false

 队列的基本操作：

q.push(item);       //将item压入队尾
q.pop();            //删除队首的元素，但不会返回  
q.front();          //返回队首的元素，但不会删除
q.back();           //返回队尾的元素，但不会删除
q.size();           //返回队列元素的个数  
q.empty();          //检查队列是否为空，如果为空返回true，否则返回false

 解题方法：

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

    int pop() {
        if(stack2.size() <= 0){   //如果stack2中有元素，那么不可以执行。
            while(stack1.size() > 0){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        int res;
        if(stack2.size() > 0){
            res = stack2.top();
            stack2.pop();
        }
        return res;
    }

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