用两个栈实现队列功能【剑指offer】

题目描述：

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

程序代码：

【方法一】

 class Solution
 {
 public:

     void push(int node) {
         stack1.push(node);

     }

     int pop() {
         int a;
         if (stack2.size() <= ){
             while (stack1.size()>){
                 a = stack1.top();
                 stack2.push(a);
                 stack1.pop();
             }
         }
         a = stack2.top();
         stack2.pop();

         return a;

     }

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

【方法二】

 class Solution
 {
 public:
     int cou = ;
     void push(int node) {
         stack1.push_back(node);
         stack2.push_back(cou++);
     }

     int pop() {
         int i = ;
         while (stack2[i] == -)
         {
             i++;
         }
         stack2[i] = -;
         return stack1[i];
     }

 private:
     vector <int> stack1;//存数
     vector <int> stack2;//地址
 };

测试用例：

 int main()
 {
     Solution fun;
     int a[] = {,,,,,,,,,};
     int i;
     for ( i = ; i < ; i++)
     {
         fun.push(a[i]);
     }
     for ( i = ; i < ; i++)
     {
         cout << " " << fun.pop() /*<< endl*/;
     }
     getchar();
     return ;
 }