239.sliding-window-maximum.md 5.6 KB
Newer Older
Y
yulecc 已提交
1
## 题目地址(239. 滑动窗口最大值)
L
luzhipeng 已提交
2

Y
yulecc 已提交
3
https://leetcode-cn.com/problems/sliding-window-maximum/
L
luzhipeng 已提交
4 5 6 7

## 题目描述

```
Y
yulecc 已提交
8
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
L
luzhipeng 已提交
9

Y
yulecc 已提交
10
返回滑动窗口中的最大值。
L
luzhipeng 已提交
11

Y
yulecc 已提交
12
 
L
luzhipeng 已提交
13

Y
yulecc 已提交
14 15 16 17 18 19 20 21 22
进阶:

你能在线性时间复杂度内解决此题吗?

 

示例:

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
L
lucifer 已提交
23 24
输出: [3,3,5,5,6,7]
解释:
Y
yulecc 已提交
25 26

  滑动窗口的位置                最大值
L
luzhipeng 已提交
27 28 29 30 31 32 33
---------------               -----
[1  3  -1] -3  5  3  6  7       3
 1 [3  -1  -3] 5  3  6  7       3
 1  3 [-1  -3  5] 3  6  7       5
 1  3  -1 [-3  5  3] 6  7       5
 1  3  -1  -3 [5  3  6] 7       6
 1  3  -1  -3  5 [3  6  7]      7
Y
yulecc 已提交
34 35 36 37 38 39 40
 

提示:

1 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
1 <= k <= nums.length
L
luzhipeng 已提交
41 42 43

```

L
lspeer 已提交
44 45 46 47
## 前置知识

- 队列
- 滑动窗口
L
lucifer 已提交
48

49 50 51 52 53 54 55
## 公司

- 阿里
- 腾讯
- 百度
- 字节

L
luzhipeng 已提交
56 57 58
## 思路

符合直觉的想法是直接遍历 nums, 然后然后用一个变量 slideWindow 去承载 k 个元素,
L
lucifer 已提交
59
然后对 slideWindow 求最大值,这是可以的,遍历一次的时间复杂度是 $N$,k 个元素求最大值时间复杂度是 $k$, 因此总的时间复杂度是 O(n \* k).代码如下:
L
luzhipeng 已提交
60

L
lucifer 已提交
61 62
JavaScript:

L
luzhipeng 已提交
63
```js
L
lucifer 已提交
64
var maxSlidingWindow = function (nums, k) {
L
luzhipeng 已提交
65 66 67 68 69 70 71 72 73 74 75 76 77 78
  // bad 时间复杂度O(n * k)
  if (nums.length === 0 || k === 0) return [];
  let slideWindow = [];
  const ret = [];
  for (let i = 0; i < nums.length - k + 1; i++) {
    for (let j = 0; j < k; j++) {
      slideWindow.push(nums[i + j]);
    }
    ret.push(Math.max(...slideWindow));
    slideWindow = [];
  }
  return ret;
};
```
L
lucifer 已提交
79

L
lucifer 已提交
80 81 82 83 84 85 86 87 88 89 90
Python3:

```python
class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if k == 0: return []
        res = []
        for r in range(k - 1, len(nums)):
            res.append(max(nums[r - k + 1:r + 1]))
        return res
```
L
luzhipeng 已提交
91 92 93

但是如果真的是这样,这道题也不会是 hard 吧?这道题有一个 follow up,要求你用线性的时间去完成。

L
lucifer 已提交
94 95 96 97 98 99 100 101 102 103 104
其实,我们没必须存储窗口内的所有元素。 如果新进入的元素比前面的大,那么前面的元素就不再有利用价值,可以直接移除。这提示我们使用一个[单调递增栈](../thinkings/monotone-stack.md "单调栈专题")来完成。

但由于窗口每次向右移动的时候,位于窗口最左侧的元素是需要被擦除的,而栈只能在一端进行操作。

而如果你使用数组实现,就是可以在另一端操作了,但是时间复杂度仍然是 $O(k)$,和上面的暴力算法时间复杂度一样。

因此,我们考虑使用链表来实现,维护两个指针分别指向头部和尾部即可,这样做的时间复杂度是 $O(1)$,这就是双端队列。

因此思路就是用一个双端队列来保存`接下来的滑动窗口可能成为最大值的数`

具体做法:
L
luzhipeng 已提交
105

L
lucifer 已提交
106
- 入队列
L
luzhipeng 已提交
107 108
- 移除失效元素,失效元素有两种

L
lucifer 已提交
109 110
1. 一种是已经超出窗口范围了,比如我遍历到第 4 个元素,k = 3,那么 i = 0 的元素就不应该出现在双端队列中了
   具体就是`索引大于 i - k + 1的元素都应该被清除`
L
luzhipeng 已提交
111 112 113

2. 小于当前元素都没有利用价值了,具体就是`从后往前遍历(双端队列是一个递减队列)双端队列,如果小于当前元素就出队列`

L
lucifer 已提交
114
经过上面的分析,不难知道双端队列其实是一个递减的一个队列,因此队首的元素一定是最大的。用图来表示就是:
L
luzhipeng 已提交
115

L
lucifer 已提交
116
![](https://tva1.sinaimg.cn/large/007S8ZIlly1ghltxg29buj30hb0di757.jpg)
L
luzhipeng 已提交
117 118 119

## 关键点解析

L
luzhipeng 已提交
120 121 122
- 双端队列简化时间复杂度

- 滑动窗口
L
luzhipeng 已提交
123 124 125

## 代码

L
lucifer 已提交
126 127
JavaScript:

L
lucifer 已提交
128 129
JS 的 deque 实现我这里没有写, 大家可以参考 [collections/deque](https://github.com/montagejs/collections/blob/master/deque.js)

L
luzhipeng 已提交
130
```js
L
lucifer 已提交
131
var maxSlidingWindow = function (nums, k) {
L
luzhipeng 已提交
132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
  // 双端队列优化时间复杂度, 时间复杂度O(n)
  const deque = []; // 存放在接下来的滑动窗口可能成为最大值的数
  const ret = [];
  for (let i = 0; i < nums.length; i++) {
    // 清空失效元素
    while (deque[0] < i - k + 1) {
      deque.shift();
    }

    while (nums[deque[deque.length - 1]] < nums[i]) {
      deque.pop();
    }

    deque.push(i);

    if (i >= k - 1) {
      ret.push(nums[deque[0]]);
    }
  }
  return ret;
};
```

L
lucifer 已提交
155 156
**复杂度分析**

L
lucifer 已提交
157 158
- 时间复杂度:$O(N * k)$,如果使用双端队列优化的话,可以到 $O(N)$
- 空间复杂度:$O(k)$
L
lucifer 已提交
159

L
lucifer 已提交
160 161 162 163 164
Python3:

```python
class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
L
lucifer 已提交
165 166 167 168 169 170 171 172
        q = collections.deque() # 本质就是单调队列
        ans = []
        for i in range(len(nums)):
            while q and nums[q[-1]] <= nums[i]: q.pop() # 维持单调性
            while q and i - q[0] >= k: q.popleft() # 移除失效元素
            q.append(i)
            if i >= k - 1: ans.append(nums[q[0]])
        return ans
L
lucifer 已提交
173 174
```

L
lucifer 已提交
175 176
**复杂度分析**

L
lucifer 已提交
177 178
- 时间复杂度:$O(N)$
- 空间复杂度:$O(k)$
L
lucifer 已提交
179

L
luzhipeng 已提交
180 181 182
## 扩展

### 为什么用双端队列
L
lucifer 已提交
183

L
lucifer 已提交
184
因为删除无效元素的时候,会清除队首的元素(索引太小了)或者队尾(元素太小了)的元素。 因此需要同时对队首和队尾进行操作,使用双端队列是一种合乎情理的做法。
L
lucifer 已提交
185 186 187 188

大家对此有何看法,欢迎给我留言,我有时间都会一一查看回答。更多算法套路可以访问我的 LeetCode 题解仓库:https://github.com/azl397985856/leetcode 。 目前已经 37K star 啦。
大家也可以关注我的公众号《力扣加加》带你啃下算法这块硬骨头。
![](https://tva1.sinaimg.cn/large/007S8ZIlly1gfcuzagjalj30p00dwabs.jpg)