剑指offer——07用两个栈实现队列

题目描述

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

 

题解：

　　有两个栈，stack1和stack2

　　push只在stack1上操作，pop只在stack2上操作

　　当stack2不为空时，直接从stack2栈顶弹出

　　当stack2为空时，将stack1的数据全部弹出压入stack2中，然后再从stack2中弹出。

 

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

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

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