LintCode 40.用栈实现队列

描述：

正如标题所述，你需要使用两个栈来实现队列的一些操作。

队列应支持push(element)，pop() 和 top()，其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例：

Example 1:

Input:
    push(1)
    pop()    
    push(2)
    push(3)
    top()    
    pop()     
Output:
    1
    2
    2

Example 2:

Input:
    push(1)
    push(2)
    push(2)
    push(3)
    push(4)
    push(5)
    push(6)
    push(7)
    push(1)
Output:
[]

挑战

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

代码：

public class MyQueue {
    Stack<Integer> stack1=new Stack<Integer>();
    Stack<Integer> stack2=new Stack<Integer>();
    public MyQueue() {
        // do intialization if necessary
    }

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

    /*
     * @return: An integer
     */
    public int pop() {
        // write your code here
        int data;
        if(!stack2.empty()){
            data=stack2.peek();
            stack2.pop();
        }
        else{
            while(!stack1.empty()){
                int tmp=stack1.peek();
                stack1.pop();
                stack2.push(tmp);
            }
            data=stack2.peek();
            stack2.pop();//忘记弹出了
        }
        return data;
    }

    /*
     * @return: An integer
     */
    public int top() {
        // write your code here
        int data;
        if(!stack2.empty()){
            data=stack2.peek();
            //stack2.pop();
        }
        else{
            while(!stack1.empty()){
                int tmp=stack1.peek();
                stack1.pop();
                stack2.push(tmp);
            }
            data=stack2.peek();
        }
        return data;
    }
}

补充说明：

这道题是《剑指Offer》的面试题7。

删除一个元素的步骤：当stack2中不为空时，在stack2中的栈顶元素是最先进入队列的元素，可以弹出。如果stack2为空时，我们把stack1中的元素逐个弹出并压入stack2。由于先进入队列的元素被压到stack1的底端，经过弹出和压入之后就处于stack2的顶端了，又可以直接弹出。

压入即压入stack1即可。