剑指offer-----用两个栈实现队列(java版)

一 题目

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

二 例子

测试操作：["PUSH1","PUSH2","PUSH3","POP","POP","PUSH4","POP","PUSH5","POP","POP"]

对应输出：1 2 3 4 5

三 思路

一开始push的时候只对 栈1 push，在pop操作时，遍历栈1，将栈1元素逐个弹出然后push到栈2，这时将栈2执行pop操作，就可以变成是队列的FIFO的形式，但是要注意一点是，我们在执行pop操作时，是把栈1元素全部pop，然后再push到栈2的（这时栈1是空的），所以在进行一次pop操作后，需要将栈2剩余的元素push回栈1中，这样下次pop操作才可以得到正确的结果。

四 程序源码

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();

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

    public int pop() {
    	while(!stack1.empty()){
     	   stack2.push(stack1.pop());
        }
    	int first = stack2.pop();
    	while(!stack2.empty()){ //将栈2剩余元素push回栈1，同时栈2弹出，否则下次结果会不一致
    	   stack1.push(stack2.pop());
    	}
       return first;
    }
}