【栈】用两个栈实现队列

 

大致思路：

push都是入栈1

要pop的时候，要先从栈1将所有数据都转入栈2，出栈一个后要把其余数据都返回栈1。

这样才能保证每次出栈都是出的最先入栈的元素，也就是栈1栈底的元素。

 

AC代码：

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

    int pop() {
        //全部都入栈2
        while(stack1.empty()==false)
        {
            stack2.push(stack1.top());
            stack1.pop();
        }
        int a = stack2.top();
        stack2.pop();
        //弹栈后再全部重新返回栈1
        while(stack2.empty()==false)
        {
            stack1.push(stack2.top());
            stack2.pop();
        }
        return a;
    }

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