今日题目:
今天做了“用 stack 实现 queue”和“用 queue 实现 stack”的两道题目,虽然难度不大,但是用 queu 实现 stack 的思路没有想到,而且有点别扭,这个要格外记一下。
LC 232. 用栈实现队列 【easy】
这个题目很容易了,使用两个 stack 来实现 queue。
在 Java 中,可以使用 ArrayList 或者 Dequeue 来当作 stack。
LC 225. 用队列实现栈 【思路没想到】 ⭐⭐⭐
用 queue 来实现 stack 有一点别扭了。
实现思路可以参考官方题解。使用了两个 queue:
- queue1:所引用的 queue 维护了栈弹出的顺序,即越新的元素越在前面。所以在实现 pop() 操作时,只需要让 queue1 的队首出队即可。
- queue2:平时为空队列,当需要添加新元素时,把新元素放入这个空队列,然后让 queue1 的所有元素依次出队并放入 queue2 这个队列,这样就实现了最新的元素在队首的顺序。完成后,queue1 变空队列了,然后交换 queue1 与 queue2 的变量引用即可。
class MyStack {
private LinkedList<Integer> queue1; // 数据 poll 的顺序就是 stack 的顺序
private LinkedList<Integer> queue2; // 一直保持为空
public MyStack() {
queue1 = new LinkedList<>();
queue2 = new LinkedList<>();
}
public void push(int x) {
queue2.offer(x);
while (!queue1.isEmpty()) {
queue2.offer(queue1.poll());
}
// swap queue1 and queue2
var temp = queue1;
queue1 = queue2;
queue2 = temp;
}
public int pop() {
return queue1.poll();
}
public int top() {
return queue1.peek();
}
public boolean empty() {
return queue1.isEmpty();
}
}