题目描述

image.png

解题思路

对于普通队列,入队 push_back() 和出队 pop_front() 的时间复杂度均为 O(1) ;本题难点为实现查找最大值 max_value() 的 O(1) 时间复杂度。 假设队列中存储 N 个元素,从中获取最大值需要遍历队列,时间复杂度为 O(N) ,单从算法上无优化空间。

如下图所示,最直观的想法是 维护一个最大值变量 ,在元素入队时更新此变量即可;但当最大值出队后,并无法确定下一个 次最大值 ,因此不可行。
所以需要一个递减的队列来维护最大值,如果取最大值就直接返回,此时时间复杂度O(1),就是空间换时间。
为了实现此递减列表,需要使用 双向队列 ,假设队列已经有若干元素:

  • 当执行入队 push_back() 时: 若入队一个比队列某些元素更大的数字 xx ,则为了保持此列表递减,需要将双向队列 尾部所有小于 xx 的元素 弹出。
  • 当执行出队 pop_front() 时: 若出队的元素是最大元素,则 双向队列 需要同时 将首元素出队 ,以保持队列和双向队列的元素一致性。

使用双向队列原因:维护递减列表需要元素队首弹出、队尾插入、队尾弹出操作皆为 O(1) 时间复杂度。

image.png
使用大小堆是不符合此题,需要自定义队列,自定义的队列和滑动窗口最大值一致。

  1. class MaxQueue {
  2. Deque<Integer> quque;
  3. MyQueue myQueue;
  4. public MaxQueue() {
  5. quque = new LinkedList<>();
  6. myQueue = new MyQueue();
  7. }
  8. public int max_value() {
  9. if (quque.isEmpty()) return -1;
  10. return myQueue.peek();
  11. }
  12. public void push_back(int value) {
  13. quque.add(value);
  14. myQueue.push(value);
  15. }
  16. public int pop_front() {
  17. if (quque.isEmpty()) return -1;
  18. myQueue.poll(quque.peek());
  19. return quque.poll();
  20. }
  21. class MyQueue {
  22. Deque<Integer> deque = new LinkedList<>();
  23. void push(int val) {
  24. while (!deque.isEmpty() && deque.getLast() < val) {
  25. deque.removeLast();
  26. }
  27. deque.add(val);
  28. }
  29. void poll(int val) {
  30. if (!deque.isEmpty() && deque.peek() == val) {
  31. deque.poll();
  32. }
  33. }
  34. int peek() {
  35. return deque.peek();
  36. }
  37. }
  38. }