用栈实现队列-栈和队列学习

用栈实现队列-栈和队列学习

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

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

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

分析

栈是先进后出，队列是先进先出。
第一想法是用栈一接收push的数。每次pop和top的时候，把栈一的数转移到栈二去，处理完再转回来。
后面想想怎么优化
首先想到的是，其实栈二的数不用立即转回去，等push的判断一下，如果栈一为空再转。
后面仔细一想其实不用转回去
push的时候将数打入栈一
pop和top的时候将栈一的数打入栈二即可，因为这样顺序也永远是正确的。

代码

Stack<Integer> stack_1;
Stack<Integer> stack_2;
public MyQueue() {
// do intialization if necessary
stack_1 = new Stack<Integer>();
stack_2 = new Stack<Integer>();
}

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

/*
 * @return: An integer
 */
public int pop() {
    // write your code here
    while (!stack_1.isEmpty()) {
        stack_2.push(stack_1.pop());
    }
    return stack_2.pop();
}

/*
 * @return: An integer
 */
public int top() {
    // write your code here
    while (!stack_1.isEmpty()) {
        stack_2.push(stack_1.pop());
    }
    return stack_2.peek();
}