用栈实现队列 -LintCode

正如标题所述，你需要使用两个栈来实现队列的一些操作。
队列应支持push(element)，pop() 和 top()，其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。

样例
比如push(1), pop(), push(2), push(3), top(), pop()，你应该返回1，2和2
挑战
仅使用两个栈来实现它，不使用任何其他数据结构，push，pop 和 top的复杂度都应该是均摊O(1)的

思路：
利用两个栈来实现队列，当push时，元素压入stack1，当进行pop或top操作时，取stack2的栈顶或出栈，若此时stack2为空，将stack1的元素出栈并压入stack2中(保证先进先出)，直至stack1为空。

#ifndef C40_H
#define C40_H
#include<iostream>
#include<stack>
using namespace std;
class MyQueue {
public:
    MyQueue() {
        // do intialization if necessary
    }

    /*
    * @param element: An integer
    * @return: nothing
    */
    void push(int element) {
        // write your code here
        stack1.push(element);//向stack1中压栈
    }

    /*
    * @return: An integer
    */
    int pop() {
        // write your code here
        int val = 0;
        //如果stack2中存在元素,返回stack2的栈顶,之后出栈
        //否则,将stack1中的元素依次出栈,并压入stack2中,返回stack2的栈顶,之后出栈
        if (stack2.empty())
        {
            while (!stack1.empty())
            {
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        val = stack2.top();
        stack2.pop();
        return val;
    }

    /*
    * @return: An integer
    */
    int top() {
        // write your code here
        //如果stack2中存在元素,返回stack2的栈顶
        //否则,将stack1中的元素依次出栈,并压入stack2中,之后返回stack2的栈顶
        if (stack2.empty())
        {
            while (!stack1.empty())
            {
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        return stack2.top();
    }
    stack<int> stack1, stack2;
};
#endif