【C++】单调队列 详解

今天我们来讲一下单调队列与栈。

这两种数据结构虽然没有在c++的stl中有直接的实现，但是在做题过程中，很容易有单调队列（栈）的使用，尤其是在一些比较难的题目中。

单调队列

1.1 单调队列介绍

单调队列 是 队列中元素之间的关系具有单调性，而且，队首和队尾都可以进行出队操作，只有队尾可以进行入队操作，本质是有双端队列deque实现的。

---

我们以 单调递减的队列 为例子：

先打个比方：假如你在饭堂打饭时，有个人人高马大，急匆匆跑过来，看排了这么一长串队，心中急躁，从队列最后的一个人开始，看见好欺负的就赶走，自己站着，直到干不过的就停下，这就是单调递减队列。也就是允许两端弹出，只允许一端插入的队列（允许两端插入，只允许一端弹出的也属于双端队列）。

我们以 7 6 8 12 9 10 3 例子建立一个单调递减队列：

---

同理，如果是 建立一个单调递增队列：

我们也可换一个比方：排队的时候，年龄小的孩子要优先，如果这个小孩子发现前面的人比他大，就把他赶出队列，直到前面的那个人比他年龄还小，他就不动了。

显然，最后单调队列的对头不是最大元素，就是最小的元素。显然，如果只是想找数列中的极值得话，完全没必要用一个队列让元素“进进出出”，大家脑子都晕了。

以本人的拙见看，单调队列一般用来在一个动态小区间中寻找极值。

接下来我们直接通过例题来感受一下。

---

1.2 单调队列的应用

1.2.1 滑动窗口最大值

1.2.1.1 题目描述

leetcode 题目链接

题目：给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。返回 滑动窗口中的最大值

事例1：

输入：nums = [1,3,-1,-3,5,3,6,7], k = 3
输出：[3,3,5,5,6,7]
解释：
滑动窗口的位置                最大值
---------------               -----
[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

事例2：

输入：nums = [1], k = 1
输出：[1]

---

1.2.1.2 结题思路

这题其实有三种做法，这里我们先不讲单调队列的做法，现讲优先级队列大的方法。
原因在于【单调队列方法】实际上是【优先级队列方法】的一种优化，直接讲单调队列难度较大，不好理解。

ps:这里我们直接跳过遍历每个小窗口的高复杂度（O(nk)）做法。

方法一（优先队列）

对于「最大值」，我们可以想到一种非常合适的数据结构，那就是优先队列（堆），其中的大根堆可以帮助我们实时维护一系列元素中的最大值。

在c++中 底层用堆实现的适配器 是priority_queue（优先级队列）。不了解的同学可以看下面这篇博客：
【C++】手把手教你写出自己的Stack和Queue类

我们以数组【1,9,-1,-3,5,3,6,7】（k=3）为例子来讲解，我先将步骤图放在下面：

• 初始时，我们将数组 \textit{nums}nums 的前 kk 个元素放入优先队列中【上图中的1步骤】。
• 每当我们向右移动窗口时，我们就可以把一个新的元素放入优先队列中【上图中的2,3,4,5,6步】，此时堆顶的元素就是堆中所有元素的最大值 。但是这个堆中的最大值可能不是我们想要的，比如【上图中第3步】，我们将9入队，此时9为最值，但显然9都不在区间内，显然不是当前区间最大值。在这种情况下，这个值在数组中的位置一定出现在滑动窗口左边界的左侧。（即9在-1的左边），因此，当我们后续继续向右移动窗口时，这个值只会里滑动窗口越来越远，我们可以将其永久地从优先队列中移除。
• 我们不断地移除堆顶的元素，直到其确实出现在滑动窗口中。此时，堆顶元素就是滑动窗口中的最大值。为了方便判断堆顶元素与滑动窗口的位置关系，我们在优先队列需要同时存储元素本身和其下标。

---

这里为什么要说“不断地移除堆顶的元素，直到其确实出现在滑动窗口中”，这是因为，有时候可能会出现几个队列内元素实际上都不在滑动窗口的情况：

比如说【1,9,9,-3,5,3,6,7】，在下图的第4步，我们删了两次队头元素才结束。

---

实现代码

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        int n = nums.size();
        priority_queue<pair<int, int>> q;
        for (int i = 0; i < k; ++i) {
            q.emplace(nums[i], i);
        }
        vector<int> ans = {q.top().first};
        for (int i = k; i < n; ++i) {
            q.emplace(nums[i], i);
            while (q.top().second <= i - k) {
                q.pop();
            }
            ans.push_back(q.top().first);
        }
        return ans;
    }
};

复杂度分析：

• 时间复杂度：O(nlogn)，其中 nn 是数组 nums 的长度。在最坏情况下，数组 nums 中的元素单调递增，那么最终优先队列中包含了所有元素，没有元素被移除。由于将一个元素放入优先队列的时间复杂度为O(logn)，因此总时间复杂度为O(nlogn)。

• 空间复杂度：O(n)，即为优先队列需要使用的空间。这里所有的空间复杂度分析都不考虑返回的答案需要的 O(n) 空间，只计算额外的空间使用。

---

方法二 （单调队列）

我们可以顺着方法一的思路继续进行优化。

我们知道，优先级队列的关键是对一个窗口内的元素用大堆排降序，但是实际上，队列没有必要维护窗口中的所有元素。只需要维护有可能成为窗口中最大值的元素即可，同时保持其单调递减性。

这是什么意思？ 我们举个例子，在一个移动窗口内，有两个数字满足 nums[i]<nums[j] ,且i<j。那么当滑动窗口向右移动时，只要 i还在窗口中，那么 j一定也还在窗口中，这是 i在 j 的左侧所保证的。同时，由于 nums[i]<nums[j]，nums[i]也不不可能成为最大值了。那么之前把num[i]留在队列里就失去了价值。

我们用一个窗口 【2 3 5 1 4】将两种方式对比。

打个不恰当的比方，一个国家有几个皇储，但是其中有几个根本没有一点机会，所以干脆退出竞争。宫廷也不用浪费资源培养他们了。

---

了解了大致的优化思路之后，我们来看一下过程：

老样子，看图说话，我们以【2 3 5 1 4 8 9 0】(k=5)这个数组为例子：

【纠正】上图中最后的0应该入队，存在笔误，感谢@愛ななせ 指出

• 当滑动窗口向右移动时，我们需要把一个新的元素放入队列中。为了保持队列的性质，我们会不断地将新的元素与队尾的元素相比较，如果前者大于等于后者，那么队尾的元素就可以被永久地移除，我们将其弹出队列。我们需要不断地进行此项操作，直到队列为空或者新的元素小于队尾的元素。
• 由于队列中下标对应的元素是严格单调递减的，因此此时队首下标对应的元素就是滑动窗口中的最大值。但与方法一中相同的是，此时的最大值可能在滑动窗口左边界的左侧，并且随着窗口向右移动，它永远不可能出现在滑动窗口中了。因此我们还需要不断从队首弹出元素，直到队首元素在窗口中为止。（在图上没体现）

所以说，这个单调队列还要保证为了可以同时弹出队首和队尾的元素，所以要是用双端队列deque去实现。

---

代码实现

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        int n = nums.size();
        deque<int> q;
        //使用第一个窗口去初始化单调队列
        for (int i = 0; i < k; ++i) {
            while (!q.empty() && nums[i] >= nums[q.back()]) {
                q.pop_back();
            }
            q.push_back(i);//队列存储的是下标
        }
        //ans存储每个窗口的最大值
        vector<int> ans = {nums[q.front()]};
        
        for (int i = k; i < n; ++i) {
            while (!q.empty() && nums[i] >= nums[q.back()]) {
                q.pop_back();
            }
            q.push_back(i);
            //检查最大元素是否在窗口区间里
            if (q.front() <= i - k) {
                q.pop_front();
            }
            ans.push_back(nums[q.front()]);
        }
        return ans;
    }
};

复杂度分析：

• 时间复杂度：O(n)，其中 n是数组 \nums 的长度。每一个下标恰好被放入队列一次，并且最多被弹出队列一次，因此时间复杂度为 O(n)O(n)。

• 空间复杂度：O(k)。与方法一不同的是，在方法二中我们使用的数据结构是双向的，因此「不断从队首弹出元素」保证了队列中最多不会有超过 k+1 个元素，因此队列使用的空间为 O(k)

---

1.3 单调队列的模拟实现

看了上面的题，其实我们也可以自己实现一个MyQueue 类，相当于一个容器适配器。

设计单调队列，push()和pop()要保持如下的规则：

1. push(value) :如果push的元素value大于队尾的数值，那就将队尾的元素弹出直到value小于或等于入口元素的数值，或者直到队列为空。
2. pop() : 如果窗口要移除的元素value等于单调队列的队头元素，那么队列弹出队头元素 ，否则不做任何操作。这个表述虽然和上文题目中有点不一样，但是意思是相同的。

下面贴出myQueue的代码：

#pragma once
namespace yyk {
	template<class T,class Compare=greater<T>>
	class myQueue {

	public:
	
		void push(T value) {
			Compare com;
			while (!que.empty() && com(value , que.back())) {
				que.pop_back();
			}
			que.push_back(value);
		}

		void pop(T value) {
			if (!que.empty() && value == que.front()) {
				que.pop_front();
			}
		}

		const T& front() {
			return que.front();
		}
		bool empty() {
			return que.empty();
		}

	private:
		deque<T> que;
	};
}

---

同时我们也可以用我们模拟实现的单调栈来实现【滑动窗口最大值】

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
#include"MyQueue.h"

void Print(yyk::myQueue<int> que) {
	while (!que.empty()) {
		cout << que.front() << " ";
		que.pop(que.front());
	}
	cout << endl;
}
vector<int>maxSlidingWindow(vector<int>& nums, int k) {
	yyk::myQueue<int> que;
	vector<int>ans;

	for (int i = 0; i < k; i++) {
		que.push(nums[i]);
	}
	ans.push_back(que.front());
	Print(que);

	for (int i = k; i < nums.size(); ++i) {
		que.pop(nums[i - k]);
		que.push(nums[i]);
		ans.push_back(que.front());
		Print(que);
	}

	return ans;
}

int main() {
	vector<int>v = { 1,9,9,-3,5,3,6,7 };
	maxSlidingWindow(v, 3);
}

如果这时候题目要我们求最小值也很简单，我们只要在实例化myQueue的时候，加上第二个模板参数就行了。

yyk::myQueue<int，Less<int>> que;

---

1.4 其他例题

之后补充。