用队列实现栈还是有点别扭。

题目

力扣题目链接

请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。

实现 MyStack 类:

  • void push(int x) 将元素 x 压入栈顶。
  • int pop() 移除并返回栈顶元素。
  • int top() 返回栈顶元素。
  • boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。

注意:

  • 你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。
  • 你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。

示例:

  1. 输入:
  2. ["MyStack", "push", "push", "top", "pop", "empty"]
  3. [[], [1], [2], [], [], []]
  4. 输出:
  5. [null, null, null, 2, 2, false]
  6. 解释:
  7. MyStack myStack = new MyStack();
  8. myStack.push(1);
  9. myStack.push(2);
  10. myStack.top(); // 返回 2
  11. myStack.pop(); // 返回 2
  12. myStack.empty(); // 返回 False

提示:

  • 1 <= x <= 9
  • 最多调用100 次 push、pop、top 和 empty
  • 每次调用 pop 和 top 都保证栈不为空

进阶:你能否实现每种操作的均摊时间复杂度为 O(1) 的栈?换句话说,执行 n 个操作的总时间复杂度 O(n) ,尽管其中某个操作可能需要比其他操作更长的时间。你可以使用两个以上的队列。

思路

(这里要强调是单向队列)

有的同学可能疑惑这种题目有什么实际工程意义,其实很多算法题目主要是对知识点的考察和教学意义远大于其工程实践的意义,所以面试题也是这样。

按照题目要求,我们要使用两个队列来模拟一个后入先出(LIFO)的栈。
225.用队列实现栈.gif
用两个队列 que1 和 que2 实现队列的功能,que2 其实完全就是一个备份的作用,把 que1 最后面的元素以外的元素都备份到 que2 ,然后弹出最后面的元素,再把其他元素从 que2 导回 que1 。

再进一步,其实可以不用导回 que1 的,直接交换 que1 和 que2 就可以了。

优化

其实这道题目就是用一个队列就够了。

一个队列在模拟栈弹出元素的时候只要将队列头部的元素(除了最后一个元素外) 重新添加到队列尾部,此时在去弹出元素就是栈的顺序了。

答案

Java

使用两个 Queue 实现

  1. class MyStack {
  2. Queue<Integer> queue1; // 和栈中保持一样元素的队列
  3. Queue<Integer> queue2; // 辅助队列
  4. /** Initialize your data structure here. */
  5. public MyStack() {
  6. queue1 = new LinkedList<>();
  7. queue2 = new LinkedList<>();
  8. }
  9. /** Push element x onto stack. */
  10. public void push(int x) {
  11. queue2.offer(x); // 先放在辅助队列中
  12. while (!queue1.isEmpty()){
  13. queue2.offer(queue1.poll());
  14. }
  15. Queue<Integer> queueTemp;
  16. queueTemp = queue1;
  17. queue1 = queue2;
  18. queue2 = queueTemp; // 最后交换queue1和queue2,将元素都放到queue1中
  19. }
  20. /** Removes the element on top of the stack and returns that element. */
  21. public int pop() {
  22. return queue1.poll(); // 因为queue1中的元素和栈中的保持一致,所以这个和下面两个的操作只看queue1即可
  23. }
  24. /** Get the top element. */
  25. public int top() {
  26. return queue1.peek();
  27. }
  28. /** Returns whether the stack is empty. */
  29. public boolean empty() {
  30. return queue1.isEmpty();
  31. }
  32. }

使用两个 Deque 实现

  1. class MyStack {
  2. // Deque 接口继承了 Queue 接口
  3. // 所以 Queue 中的 add、poll、peek等效于 Deque 中的 addLast、pollFirst、peekFirst
  4. Deque<Integer> que1; // 和栈中保持一样元素的队列
  5. Deque<Integer> que2; // 辅助队列
  6. /** Initialize your data structure here. */
  7. public MyStack() {
  8. que1 = new ArrayDeque<>();
  9. que2 = new ArrayDeque<>();
  10. }
  11. /** Push element x onto stack. */
  12. public void push(int x) {
  13. que1.addLast(x);
  14. }
  15. /** Removes the element on top of the stack and returns that element. */
  16. public int pop() {
  17. int size = que1.size();
  18. size--;
  19. // 将 que1 导入 que2 ,但留下最后一个值
  20. while (size-- > 0) {
  21. que2.addLast(que1.peekFirst());
  22. que1.pollFirst();
  23. }
  24. int res = que1.pollFirst();
  25. // 将 que2 对象的引用赋给了 que1 ,此时 que1,que2 指向同一个队列
  26. que1 = que2;
  27. // 如果直接操作 que2,que1 也会受到影响,所以为 que2 分配一个新的空间
  28. que2 = new ArrayDeque<>();
  29. return res;
  30. }
  31. /** Get the top element. */
  32. public int top() {
  33. return que1.peekLast();
  34. }
  35. /** Returns whether the stack is empty. */
  36. public boolean empty() {
  37. return que1.isEmpty();
  38. }
  39. }

优化,使用一个 Deque 实现

  1. class MyStack {
  2. // Deque 接口继承了 Queue 接口
  3. // 所以 Queue 中的 add、poll、peek等效于 Deque 中的 addLast、pollFirst、peekFirst
  4. Deque<Integer> que1;
  5. /** Initialize your data structure here. */
  6. public MyStack() {
  7. que1 = new ArrayDeque<>();
  8. }
  9. /** Push element x onto stack. */
  10. public void push(int x) {
  11. que1.addLast(x);
  12. }
  13. /** Removes the element on top of the stack and returns that element. */
  14. public int pop() {
  15. int size = que1.size();
  16. size--;
  17. // 将 que1 导入 que2 ,但留下最后一个值
  18. while (size-- > 0) {
  19. que1.addLast(que1.peekFirst());
  20. que1.pollFirst();
  21. }
  22. int res = que1.pollFirst();
  23. return res;
  24. }
  25. /** Get the top element. */
  26. public int top() {
  27. return que1.peekLast();
  28. }
  29. /** Returns whether the stack is empty. */
  30. public boolean empty() {
  31. return que1.isEmpty();
  32. }
  33. }

REF

https://leetcode-cn.com/problems/implement-stack-using-queues/