用两个栈实现一个队列

题目

牛客网

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

解题思路

1. 用 stack1 作为 push 队列，将元素 push 到 stack1
2. 用 stack2 作为 pop 队列，当 stack2 为空时则将 stack1 的数据 push 到 stack2，否则直接 pop stack2

相当于将两个 stack 拼接：-> stack1 <::> stack2 ->

Stack<Integer> pushStack = new Stack<>();
Stack<Integer> popStack = new Stack<>();
public void push(int node) {
    pushStack.push(node);
}
public int pop() {
    if (popStack.isEmpty()) {
        while (!pushStack.isEmpty()) {
            popStack.push(pushStack.pop());
        }
    }
    if (popStack.isEmpty()) return -1;
    else return popStack.pop();
}