用两个栈实现队列

题目描述

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

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

  int pop() {
      if(stack2.empty()) {//第二个栈为空时
      while(!stack1.empty()) {//第一个栈不为空
          int num = stack1.top();//取栈顶元素
          stack1.pop();//弹出元素
          stack2.push(num);//压人第二个栈顶
        }
      }

      if(!stack2.empty()) {
        int num = stack2.top();
        stack2.pop();
        return num;//取第二个栈顶的元素返回
      }

      return -1;
  }

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