本题构造单调队列求解给定数组nums中大小为k的滑动窗口的最大值。主要思想为队列没有必要维护窗口里的所有元素,只需要维护有可能成为窗口里最大值的元素就可以了,同时保证队列里的元素数值是由大到小的。本题的难点点在于单调队列的实现和滑动窗口的维护。
单调队列的实现:单调队列是一种特殊的数据结构,它的特点是队列中的元素始终保持单调性。在本题中,我们需要维护一个从大到小的单调队列,这样队列的第一个元素始终是滑动窗口中的最大值。通过定义push和pop操作,我们可以高效地维护这个单调性和滑动窗口的移动。
滑动窗口的维护:在每次窗口滑动时,移除最前面的元素,同时添加一个新的元素。这个过程需要结合单调队列的原理操作,以高效地找到窗口中的最大值。
时间复杂度:O(n),空间复杂度:O(k)
思路:
- 定义一个MyQueue类,实现单调队列(从大到小)
- pop(value):若队列不为空,且需要弹出的元素是当前队列最大值,则弹出该元素
- push(value):若队列不为空,从后往前弹出比value小的元素,然后将value压入队列,因为该队列只维护最大值
- getMax():返回队列第一个元素,即窗口下的最大值
- 定义一个maxSlidingWindow函数,输入数组nums和窗口大小k
- 创建MyQueue实例que,用于存放滑动窗口中的元素
- 创建结果数组result
- 先将前k个元素放入队列
- 记录前k个元素的最大值
- 遍历数组nums,对于每个滑动窗口
- 先移除最前面的元素
- 滑动窗口加入新元素
- 记录当前窗口最大值
- 返回结果数组result
代码:
class Solution {
private:
class MyQueue { // 单调队列(从大到小)
public:
deque<int> que;
void pop(int value) {
// 不是空队列,且弹出元素需为当前队列最大值时,才弹出该元素
if (!que.empty() && value == que.front()) {
que.pop_front();
}
}
void push(int value) {
// 不是空队列,从后往前将比压入元素小的元素弹出,再压入元素,因为该队列只维护最大值
while (!que.empty() && value > que.back()) {
que.pop_back();
}
que.push_back(value);
}
// 该队列的第一个元素就是窗口下的最大值
int getMax() {
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
vector<int> result;
for (int i = 0; i < k; i++) { // 先将前k个元素放入队列
que.push(nums[i]);
}
result.push_back(que.getMax()); // 记录前k个元素的最大值
for (int i = k; i < nums.size(); i++) {
que.pop(nums[i - k]); // 在滑动窗口前,先移除最前面的元素
que.push(nums[i]); // 滑动窗口加入新元素
result.push_back(que.getMax()); // 记录当前窗口最大值
}
return result;
}
};