用两个栈实现一个队列

题目 #

牛客网

用两个栈来实现一个队列,完成队列的 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();
}