牛客网《剑指Offer》编程 5.用两个堆栈实现队列的push和pop操作

题目描述

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

解题思路

两个堆栈分别是stack1和stack2。

push操作：直接将元素push入stack1里面即可。

pop操作：将stack2作为队列首，每次有元素都从stack2中弹出即可。因此当stack2不空的时候，直接将栈顶元素弹出，就是队列首的元素；stack2为空时 ，将stack1中的元素push入stack2中，然后将stack2中 的栈顶元素弹出。

代码实现

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

    int pop() {
        int res;
        if(stack2.size()>0){
            res=stack2.top();
            stack2.pop();
        }else if(stack1.size()>0){
            while(stack1.size()>0){
                int tmp=stack1.top();
                stack1.pop();
                stack2.push(tmp);
            }
            res=stack2.top();
            stack2.pop();
        }
        return res;
    }

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