JZ5 用两个栈实现队列

原题链接

描述

---

用两个栈来实现一个队列，分别完成在队列尾部插入整数(push)和在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

示例

---

输入：["PSH1","PSH2","POP","POP"]

返回值：1,2

思路

---

把stack1用于push，stack2用于pop达到队列FIFO的效果。没有限制两个栈的大小，所以第一次pop时直接把stack1的pop顺序push进stack2，然后stack2进行pop即可。

但是第二次pop时，就不能再像刚才那样了，因为stack2如果还有元素的话，说明这些元素都比stack1栈底的元素入站早。所以再有pop时要把stack2的元素顺序pop完，才能输出stack1的元素。

解答

---

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() {
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.pop());
            }
            return stack2.pop();
        }
        return stack2.pop();
    }
}