剑指offer第五题

题目

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

思路

本来想的很简单，从stack1往stack2中一倒就把“后进先出”倒成了“先进先出”。
但是这个思路有一个问题，就是stack2里面有元素的时候。本来他应该先出的，反倒在stack2中先进，变成了后出
所以应该把stack2分成两种情况
一种它为空，用原来的思路，另一种它不为空，就直接从stack2里面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) {
       this.stack1.push(node); 
    }

    public int pop() {
        if(stack2.isEmpty()){
            while(!this.stack1.isEmpty()){
                this.stack2.push(stack1.pop());
            }
            return stack2.pop();
        }else{
            stack2.pop();
        }

    }

}