NOWCODER[编程题]——用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

给出的框架如下图，所以可以使用c++的stack，这也就意味着可以使用里面封装好的接口，就会变得更加容易一些。

解题思路

• push接口可以直接向stack1中压入数据，即stack1.push(node)；
• 队列满足FIFO的性质，我们之前把数据都压入stack1了，现在最先出来的元素应该是stack1栈底的元素，所以：如果stack2为空的话，把stack1里的元素pop出放进stack2中，直接pop即可满足题意
• 需要注意的是，给出的pop()接口是带有返回值的，所以每次操作要先取出stack1的top，把top()存进stack2中才能对stack1 进行pop() 操作

实现代码

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

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

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