JZ5 用两个栈实现队列

题目描述

点这里

思路分析

栈模拟队列。

stack1当作住栈，stack2当作副栈。
数组都压在主栈内。
push的时候直接加入栈1即可。
pop的时候把栈1元素从上到下加入栈2，弹出并返回栈2顶，然后再把栈2元素加入回到栈1即可。

代码实现

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

    int pop() {
        while(stack1.size()) stack2.push(stack1.top()),stack1.pop();
        int ans=stack2.top();
        stack2.pop();
        while(stack2.size()) stack1.push(stack2.top()),stack2.pop();
        return ans;
    }

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