1 问题
给你一个整数数组 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]
示例 3:
输入:nums = [1,-1], k = 1
输出:[1,-1]
示例 4:
输入:nums = [9,11], k = 2
输出:[11]
示例 5:
输入:nums = [4,-2], k = 2
输出:[4]
提示:
1 <= nums.length <= 105
-104 <= nums[i] <= 104
1 <= k <= nums.length
2 解法
通过构建队列,当窗口移动时,队列中的元素也同时一进一出,每次移动之后,队列告诉我们里面的最大值是什么。
这个队列应该长这个样子:
class MyQueue {
public:
void pop(int value) {
}
void push(int value) {
}
int front() {
return que.front();
}
};
每次窗口移动的时候,调用que.pop(滑动窗口中移除元素的数值),que.push(滑动窗口添加元素的数值),然后que.front()就返回我们要的最大值。
构建一个单调递减队列,设计单调队列的时候,pop,和push操作要保持如下规则:
(1)pop(value):如果窗口移除的元素value等于单调队列的出口元素,那么队列弹出元素,否则不用任何操作
(2)push(value):如果push的元素value大于入口元素的数值,那么就将队列入口的元素弹出,直到push元素的数值小于等于队列入口元素的数值为止(确保最大元素在队头)
保持如上规则,每次窗口移动的时候,只要问que.front()就可以返回当前窗口的最大值。
class Solution {
//创建一个私有队列类,队首至队尾元素依次递减
private:
class MyQueue{
public:
deque<int> que;
//用于弹出窗口需要移除的队首元素
void pop(int val)
{
//队列不为空,需要移除的窗口元素=队首元素时,则弹出队首元素
if(!que.empty() && val == que.front())
{
que.pop_front();
}
}
//每次进队时,保证队首至队尾元素依次递减
void push(int val)
{
//队列不为空,且进队元素>队尾元素,则弹出队尾元素
while(!que.empty() && val > que.back())
{
que.pop_back();
}
//进队元素<队尾元素,则进入队尾
que.push_back(val);
}
//获取队首元素
int front()
{
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> res;
//实例化单调递减队列
MyQueue que;
//放入前k个元素
for(int i = 0; i < k; i++)
{
que.push(nums[i]);
}
//队首元素即为第一个窗口的最大元素
res.push_back(que.front());
//放入后面窗口的元素
for(int i = k; i < nums.size(); i++)
{
//窗口需要移除的队首元素
que.pop(nums[i - k]);
//窗口中新进入的元素
que.push(nums[i]);
//返回窗口的最大值
res.push_back(que.front());
}
return res;
}
};
使用单调队列的时间复杂度是 O(n)。
空间复杂度因为我们定义一个辅助队列,所以是O(k)。