LintCode-40.用栈实现队列

用栈实现队列

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

样例

比如push(1), pop(), push(2), push(3), top(), pop()，你应该返回1，2和2

挑战

仅使用两个栈来实现它，不使用任何其他数据结构，push，pop 和top的复杂度都应该是均摊O(1)的

标签

栈 LintCode 版权所有 队列

code

class MyQueue {
public:
    stack<int> stack1;
    stack<int> stack2;

    MyQueue() {
        // do intialization if necessary
    }

    void push(int element) {
        // write your code here
        stack1.push(element);
    }
    
    int pop() {
        // write your code here
        int data;
        if(stack2.empty()) {
            while(!stack1.empty()) {
                data = stack1.top();
                stack2.push(data);
                stack1.pop();
            }
        }
    
        data = stack2.top();
        stack2.pop();
        return data;
    }

    int top() {
        // write your code here
        int data;
        if(stack2.empty()) {
            while(!stack1.empty()) {
                data = stack1.top();
                stack2.push(data);
                stack1.pop();
            }
        }
        else
            data = stack2.top();
        return data;
    }
};