题解 | #用两个栈实现队列#

我们使用一个栈stack1用来存入队的元素,一个栈stack2来存用于出队的队头元素,当需要出队元素的时候,我们可以先检查stack2是否为空,如果为空那么我们就将stack1中的元素全部出栈然后入栈到stack2中,然后再将stack2栈顶元素弹出,并返回即可

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

    int pop() {
        int ans = -1;
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        ans = stack2.top();
        stack2.pop();
        return ans;
    }

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