剑指Offer JZ5 用两个栈实现队列

题目描述

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

思路

1. push操作直接stack1入栈，没有问题
2. pop操作麻烦一点
   首先要将stack1中的数据按顺序出栈然后依次再入栈stack2
   这样操作后，stack2的栈顶就是我们要出队的元素

代码实现

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

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

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

注意事项

1. 当我们的stack2栈为空的时候，这时候我们的stack1入栈了一个元素5，我们就需要把stack1中的元素全部复制到我们的stack2中，然后再pop我们stack2的栈顶