单调队列.md 16.5 KB
Newer Older
L
labuladong 已提交
1 2
# 特殊数据结构:单调队列

3

L
labuladong 已提交
4 5 6 7 8





9 10
<p align='center'>
<a href="https://github.com/labuladong/fucking-algorithm" target="view_window"><img alt="GitHub" src="https://img.shields.io/github/stars/labuladong/fucking-algorithm?label=Stars&style=flat-square&logo=GitHub"></a>
L
labuladong 已提交
11
<a href="https://appktavsiei5995.pc.xiaoe-tech.com/index" target="_blank"><img class="my_header_icon" src="https://img.shields.io/static/v1?label=精品课程&message=查看&color=pink&style=flat"></a>
12 13 14 15
<a href="https://www.zhihu.com/people/labuladong"><img src="https://img.shields.io/badge/%E7%9F%A5%E4%B9%8E-@labuladong-000000.svg?style=flat-square&logo=Zhihu"></a>
<a href="https://space.bilibili.com/14089380"><img src="https://img.shields.io/badge/B站-@labuladong-000000.svg?style=flat-square&logo=Bilibili"></a>
</p>

L
labuladong 已提交
16
![](https://labuladong.github.io/algo/images/souyisou1.png)
17

L
labuladong 已提交
18
**通知:[数据结构精品课](https://aep.h5.xeknow.com/s/1XJHEO) 已更新到 V1.9,[第 11 期刷题打卡挑战(9/19 开始)](https://mp.weixin.qq.com/s/eUG2OOzY3k_ZTz-CFvtv5Q) 开始报名。**
19 20


L
labuladong 已提交
21 22 23 24 25 26 27 28

读完本文,你不仅学会了算法套路,还可以顺便解决如下题目:

| LeetCode | 力扣 | 难度 |
| :----: | :----: | :----: |
| [239. Sliding Window Maximum](https://leetcode.com/problems/sliding-window-maximum/) | [239. 滑动窗口最大值](https://leetcode.cn/problems/sliding-window-maximum/) | 🔴
| - | [剑指 Offer 59 - I. 滑动窗口的最大值](https://leetcode.cn/problems/hua-dong-chuang-kou-de-zui-da-zhi-lcof/) | 🔴
| - | [剑指 Offer 59 - II. 队列的最大值](https://leetcode.cn/problems/dui-lie-de-zui-da-zhi-lcof/) | 🟠
29 30 31

**-----------**

L
labuladong 已提交
32
前文用 [单调栈解决三道算法问题](https://labuladong.github.io/article/fname.html?fname=单调栈) 介绍了单调栈这种特殊数据结构,本文写一个类似的数据结构「单调队列」。
L
labuladong 已提交
33

L
labuladong 已提交
34
也许这种数据结构的名字你没听过,其实没啥难的,就是一个「队列」,只是使用了一点巧妙的方法,使得队列中的元素全都是单调递增(或递减)的。
L
labuladong 已提交
35

L
labuladong 已提交
36
为啥要发明「单调队列」这种结构呢,主要是为了解决下面这个场景:
L
labuladong 已提交
37

L
labuladong 已提交
38
**给你一个数组 `window`,已知其最值为 `A`,如果给 `window` 中添加一个数 `B`,那么比较一下 `A` 和 `B` 就可以立即算出新的最值;但如果要从 `window` 数组中减少一个数,就不能直接得到最值了,因为如果减少的这个数恰好是 `A`,就需要遍历 `window` 中的所有元素重新寻找新的最值**
L
labuladong 已提交
39

L
labuladong 已提交
40 41 42 43 44 45 46
这个场景很常见,但不用单调队列似乎也可以,比如优先级队列也是一种特殊的队列,专门用来动态寻找最值的,我创建一个大(小)顶堆,不就可以很快拿到最大(小)值了吗?

如果单纯地维护最值的话,优先级队列很专业,队头元素就是最值。但优先级队列无法满足标准队列结构「先进先出」的**时间顺序**,因为优先级队列底层利用二叉堆对元素进行动态排序,元素的出队顺序是元素的大小顺序,和入队的先后顺序完全没有关系。

所以,现在需要一种新的队列结构,既能够维护队列元素「先进先出」的时间顺序,又能够正确维护队列中所有元素的最值,这就是「单调队列」结构。

「单调队列」这个数据结构主要用来辅助解决滑动窗口相关的问题,前文 [滑动窗口核心框架](https://labuladong.github.io/article/fname.html?fname=滑动窗口技巧进阶) 把滑动窗口算法作为双指针技巧的一部分进行了讲解,但有些稍微复杂的滑动窗口问题不能只靠两个指针来解决,需要上更先进的数据结构。
L
labuladong 已提交
47

L
labuladong 已提交
48
比方说,你注意看前文 [滑动窗口核心框架](https://labuladong.github.io/article/fname.html?fname=滑动窗口技巧进阶) 讲的几道题目,每当窗口扩大(`right++`)和窗口缩小(`left++`)时,你单凭移出和移入窗口的元素即可决定是否更新答案。
L
labuladong 已提交
49

L
labuladong 已提交
50
但就本文开头说的那个判断一个窗口中最值的例子,你就无法单凭移出窗口的那个元素更新窗口的最值,除非重新遍历所有元素,但这样的话时间复杂度就上来了,这是我们不希望看到的。
L
labuladong 已提交
51

L
labuladong 已提交
52
我们来看看力扣第 239 题「滑动窗口最大值」,就是一道标准的滑动窗口问题:
L
labuladong 已提交
53

L
labuladong 已提交
54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
给你输入一个数组 `nums` 和一个正整数 `k`,有一个大小为 `k` 的窗口在 `nums` 上从左至右滑动,请你输出每次窗口中 `k` 个元素的最大值。

函数签名如下:

```java
int[] maxSlidingWindow(int[] nums, int k);
```

比如说力扣给出的一个示例:

![](https://labuladong.github.io/algo/images/单调队列/window.png)

接下来,我们就借助单调队列结构,用 `O(1)` 时间算出每个滑动窗口中的最大值,使得整个算法在线性时间完成。

### 一、搭建解题框架

在介绍「单调队列」这种数据结构的 API 之前,先来看看一个普通的队列的标准 API:
L
labuladong 已提交
71 72 73

```java
class Queue {
L
labuladong 已提交
74
    // enqueue 操作,在队尾加入元素 n
L
labuladong 已提交
75
    void push(int n);
L
labuladong 已提交
76
    // dequeue 操作,删除队头元素
L
labuladong 已提交
77 78 79 80
    void pop();
}
```

L
labuladong 已提交
81
我们要实现的「单调队列」的 API 也差不多:
L
labuladong 已提交
82 83 84 85 86 87 88 89 90 91 92 93 94 95

```java
class MonotonicQueue {
    // 在队尾添加元素 n
    void push(int n);
    // 返回当前队列中的最大值
    int max();
    // 队头元素如果是 n,删除它
    void pop(int n);
}
```

当然,这几个 API 的实现方法肯定跟一般的 Queue 不一样,不过我们暂且不管,而且认为这几个操作的时间复杂度都是 O(1),先把这道「滑动窗口」问题的解答框架搭出来:

L
labuladong 已提交
96 97 98 99 100 101 102 103
```java
int[] maxSlidingWindow(int[] nums, int k) {
    MonotonicQueue window = new MonotonicQueue();
    List<Integer> res = new ArrayList<>();
    
    for (int i = 0; i < nums.length; i++) {
        if (i < k - 1) {
            //先把窗口的前 k - 1 填满
L
labuladong 已提交
104
            window.push(nums[i]);
L
labuladong 已提交
105 106 107
        } else {
            // 窗口开始向前滑动
            // 移入新元素
L
labuladong 已提交
108
            window.push(nums[i]);
L
labuladong 已提交
109 110 111
            // 将当前窗口中的最大元素记入结果
            res.add(window.max());
            // 移出最后的元素
L
labuladong 已提交
112 113 114
            window.pop(nums[i - k + 1]);
        }
    }
L
labuladong 已提交
115 116 117 118 119 120
    // 将 List 类型转化成 int[] 数组作为返回值
    int[] arr = new int[res.size()];
    for (int i = 0; i < res.size(); i++) {
        arr[i] = res.get(i);
    }
    return arr;
L
labuladong 已提交
121 122 123
}
```

L
labuladong 已提交
124
![](https://labuladong.github.io/algo/images/单调队列/1.png)
L
labuladong 已提交
125 126 127 128 129

这个思路很简单,能理解吧?下面我们开始重头戏,单调队列的实现。

### 二、实现单调队列数据结构

L
labuladong 已提交
130
观察滑动窗口的过程就能发现,实现「单调队列」必须使用一种数据结构支持在头部和尾部进行插入和删除,很明显双链表是满足这个条件的。
L
labuladong 已提交
131

L
labuladong 已提交
132
「单调队列」的核心思路和「单调栈」类似,`push` 方法依然在队尾添加元素,但是要把前面比自己小的元素都删掉:
L
labuladong 已提交
133

L
labuladong 已提交
134
```java
L
labuladong 已提交
135
class MonotonicQueue {
L
labuladong 已提交
136 137 138 139 140 141 142 143 144
// 双链表,支持头部和尾部增删元素
// 维护其中的元素自尾部到头部单调递增
private LinkedList<Integer> maxq = new LinkedList<>();

// 在尾部添加一个元素 n,维护 maxq 的单调性质
public void push(int n) {
    // 将前面小于自己的元素都删除
    while (!maxq.isEmpty() && maxq.getLast() < n) {
        maxq.pollLast();
L
labuladong 已提交
145
    }
L
labuladong 已提交
146 147
    maxq.addLast(n);
}
L
labuladong 已提交
148 149 150 151
```

你可以想象,加入数字的大小代表人的体重,把前面体重不足的都压扁了,直到遇到更大的量级才停住。

L
labuladong 已提交
152
![](https://labuladong.github.io/algo/images/单调队列/3.png)
L
labuladong 已提交
153

L
labuladong 已提交
154
如果每个元素被加入时都这样操作,最终单调队列中的元素大小就会保持一个**单调递减**的顺序,因此我们的 `max` 方法可以可以这样写:
L
labuladong 已提交
155

L
labuladong 已提交
156 157 158 159
```java
public int max() {
    // 队头的元素肯定是最大的
    return maxq.getFirst();
L
labuladong 已提交
160 161 162
}
```

L
labuladong 已提交
163
`pop` 方法在队头删除元素 `n`,也很好写:
L
labuladong 已提交
164

L
labuladong 已提交
165 166 167 168 169
```java
public void pop(int n) {
    if (n == maxq.getFirst()) {
        maxq.pollFirst();
    }
L
labuladong 已提交
170 171 172
}
```

L
labuladong 已提交
173
之所以要判断 `data.getFirst() == n`,是因为我们想删除的队头元素 `n` 可能已经被「压扁」了,可能已经不存在了,所以这时候就不用删除了:
L
labuladong 已提交
174

L
labuladong 已提交
175
![](https://labuladong.github.io/algo/images/单调队列/2.png)
L
labuladong 已提交
176 177 178

至此,单调队列设计完毕,看下完整的解题代码:

L
labuladong 已提交
179 180
```java
/* 单调队列的实现 */
L
labuladong 已提交
181
class MonotonicQueue {
L
labuladong 已提交
182 183 184 185 186 187 188 189
    LinkedList<Integer> maxq = new LinkedList<>();
    public void push(int n) {
        // 将小于 n 的元素全部删除
        while (!maxq.isEmpty() && maxq.getLast() < n) {
            maxq.pollLast();
        }
        // 然后将 n 加入尾部
        maxq.addLast(n);
L
labuladong 已提交
190 191
    }
    
L
labuladong 已提交
192 193 194
    public int max() {
        return maxq.getFirst();
    }
L
labuladong 已提交
195
    
L
labuladong 已提交
196 197 198 199
    public void pop(int n) {
        if (n == maxq.getFirst()) {
            maxq.pollFirst();
        }
L
labuladong 已提交
200
    }
L
labuladong 已提交
201
}
L
labuladong 已提交
202

L
labuladong 已提交
203 204 205 206 207 208 209 210
/* 解题函数的实现 */
int[] maxSlidingWindow(int[] nums, int k) {
    MonotonicQueue window = new MonotonicQueue();
    List<Integer> res = new ArrayList<>();
    
    for (int i = 0; i < nums.length; i++) {
        if (i < k - 1) {
            //先填满窗口的前 k - 1
L
labuladong 已提交
211
            window.push(nums[i]);
L
labuladong 已提交
212 213
        } else {
            // 窗口向前滑动,加入新数字
L
labuladong 已提交
214
            window.push(nums[i]);
L
labuladong 已提交
215 216 217
            // 记录当前窗口的最大值
            res.add(window.max());
            // 移出旧数字
L
labuladong 已提交
218 219 220
            window.pop(nums[i - k + 1]);
        }
    }
L
labuladong 已提交
221 222 223 224 225 226
    // 需要转成 int[] 数组再返回
    int[] arr = new int[res.size()];
    for (int i = 0; i < res.size(); i++) {
        arr[i] = res.get(i);
    }
    return arr;
L
labuladong 已提交
227 228 229
}
```

L
labuladong 已提交
230 231 232
有一点细节问题不要忽略,在实现 `MonotonicQueue` 时,我们使用了 Java 的 `LinkedList`,因为链表结构支持在头部和尾部快速增删元素;而在解法代码中的 `res` 则使用的 `ArrayList` 结构,因为后续会按照索引取元素,所以数组结构更合适。

关于单调队列 API 的时间复杂度,读者可能有疑惑:`push` 操作中含有 while 循环,时间复杂度应该不是 `O(1)` 呀,那么本算法的时间复杂度应该不是线性时间吧?
L
labuladong 已提交
233

L
labuladong 已提交
234
这里就用到了 [算法时空复杂度分析使用手册](https://labuladong.github.io/article/fname.html?fname=时间复杂度) 中讲到的摊还分析:
L
labuladong 已提交
235

L
labuladong 已提交
236
单独看 `push` 操作的复杂度确实不是 `O(1)`,但是算法整体的复杂度依然是 `O(N)` 线性时间。要这样想,`nums` 中的每个元素最多被 `push``pop` 一次,没有任何多余操作,所以整体的复杂度还是 `O(N)`。空间复杂度就很简单了,就是窗口的大小 `O(k)`
L
labuladong 已提交
237

L
labuladong 已提交
238
### 拓展延伸
L
labuladong 已提交
239

L
labuladong 已提交
240
最后,我提出几个问题请大家思考:
L
labuladong 已提交
241

L
labuladong 已提交
242
1、本文给出的 `MonotonicQueue` 类只实现了 `max` 方法,你是否能够再额外添加一个 `min` 方法,在 `O(1)` 的时间返回队列中所有元素的最小值?
L
labuladong 已提交
243

L
labuladong 已提交
244
2、本文给出的 `MonotonicQueue` 类的 `pop` 方法还需要接收一个参数,这显然有悖于标准队列的做法,请你修复这个缺陷。
L
labuladong 已提交
245

L
labuladong 已提交
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
3、请你实现 `MonotonicQueue` 类的 `size` 方法,返回单调队列中元素的个数(注意,由于每次 `push` 方法都可能从底层的 `q` 列表中删除元素,所以 `q` 中的元素个数并不是单调队列的元素个数)。

也就是说,你是否能够实现单调队列的通用实现:

```java
/* 单调队列的通用实现,可以高效维护最大值和最小值 */
class MonotonicQueue<E extends Comparable<E>> {

    // 标准队列 API,向队尾加入元素
    public void push(E elem);

    // 标准队列 API,从队头弹出元素,符合先进先出的顺序
    public E pop();

    // 标准队列 API,返回队列中的元素个数
    public int size();

    // 单调队列特有 API,O(1) 时间计算队列中元素的最大值
    public E max();

    // 单调队列特有 API,O(1) 时间计算队列中元素的最小值
    public E min();
}
```

我将在 [单调队列通用实现及应用](https://appktavsiei5995.pc.xiaoe-tech.com/detail/i_62a692efe4b01a48520b9b9b/1) 中给出单调队列的通用实现和经典习题。
L
labuladong 已提交
272

273
**_____________**
L
labuladong 已提交
274

L
labuladong 已提交
275 276 277
**《labuladong 的算法小抄》已经出版,关注公众号查看详情;后台回复关键词「进群」可加入算法群;回复「PDF」可获取精华文章 PDF**

![](https://labuladong.github.io/algo/images/souyisou2.png)
L
labuladong 已提交
278 279


H
hzs 已提交
280 281
======其他语言代码======

B
brucecat 已提交
282 283 284
[239.滑动窗口最大值](https://leetcode-cn.com/problems/sliding-window-maximum)

### python
H
hzs 已提交
285

H
hzs 已提交
286 287 288
[SCUHZS](ttps://github.com/brucecat)提供


H
hzs 已提交
289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
```python
from collections import deque

class MonotonicQueue(object):
    def __init__(self):
        # 双端队列
        self.data = deque()

    def push(self, n):
        # 实现单调队列的push方法
        while self.data and self.data[-1] < n:
            self.data.pop()
        self.data.append(n)

    def max(self):
        # 取得单调队列中的最大值
        return self.data[0]

    def pop(self, n):
        # 实现单调队列的pop方法
        if self.data and self.data[0] == n:
            self.data.popleft()


class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        # 单调队列实现的窗口
        window = MonotonicQueue()

        # 结果
        res = []
        
        for i in range(0, len(nums)):
            
            if i < k-1:
                # 先填满窗口前k-1
                window.push(nums[i])
            else:
                # 窗口向前滑动
                window.push(nums[i])
                res.append(window.max())
                window.pop(nums[i-k+1])
        return res

B
BruceCat 已提交
333 334 335 336
```

### java

J
Justlxb0124 已提交
337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383
```java
class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int len = nums.length;
        // 判断数组或者窗口长度为0的情况
        if (len * k == 0) {
            return new int[0];
        }

        /*
        采用两端扫描的方法
        将数组分成大小为 k 的若干个窗口, 对每个窗口分别从左往右和从右往左扫描, 记录扫描的最大值
        left[] 记录从左往右扫描的最大值
        right[] 记录从右往左扫描的最大值
         */
        int[] left = new int[len];
        int[] right = new int[len];

        for (int i = 0; i < len; i = i + k) {
            // 每个窗口中的第一个值
            left[i] = nums[i];
            // 窗口的最后边界
            int index = i + k - 1 >= len ? len - 1 : i + k - 1;
            // 每个窗口的最后一个值
            right[index] = nums[index];
            // 对该窗口从左往右扫描
            for (int j = i + 1; j <= index; j++) {
                left[j] = Math.max(left[j - 1], nums[j]);
            }
            // 对该窗口从右往左扫描
            for (int j = index - 1; j >= i; j--) {
                right[j] = Math.max(right[j + 1], nums[j]);
            }
        }

        int[] arr = new int[len - k + 1];

        // 对于第 i 个位置, 它一定是该窗口从右往左扫描数组中的最后一个值, 相对的 i + k - 1 是该窗口从左向右扫描数组中的最后一个位置
        // 对两者取最大值即可
        for (int i = 0; i < len - k + 1; i++) {
            arr[i] = Math.max(right[i], left[i + k - 1]);
        }

        return arr;
    }
}
```
B
brucecat 已提交
384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442

### javascript

这里用js实现的思路和上文中一样,都是自己实现一个单调队列,注意,这里的单调队列和优先级队列(大小堆)不是同一个概念。

```js
let MonotonicQueue = function () {

    // 模拟一个deque双端队列
    this.data = [];

    // 在队尾添加元素 n
    this.push = function (n) {
        while (this.data.length !== 0 && this.data[this.data.length - 1] < n)
            this.data.pop();
        this.data.push(n);
    }

    // 返回当前队列中的最大值
    this.max = function () {
        return this.data[0];
    };

    // 队头元素如果是 n,删除它
    this.pop = function (n) {
        if (this.data.length !== 0 && this.data[0] === n)
            this.data.shift();
    };
}

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var maxSlidingWindow = function (nums, k) {
    let window = new MonotonicQueue();

    let res = []

    for (let i = 0; i < nums.length; i++) {
        if (i < k - 1) { //先把窗口的前 k - 1 填满
            window.push(nums[i]);
        } else {
            // 窗口开始向前滑动
            window.push(nums[i]);

            res.push(window.max());

            window.pop(nums[i - k + 1]);
            // nums[i - k + 1] 就是窗口最后的元素
        }
    }

    return res;

};
```