滑动窗口的最大值
滑动窗口的最大值
题目链接:https://leetcode-cn.com/problems/hua-dong-chuang-kou-de-zui-da-zhi-lcof/
首先想到的暴力解法是:先遍历一遍数组,每次比较滑动窗口内的数值大小,选出最大的
这样3,和-1又多比较了,复杂度为:O((n-k+1) k) 约等于O(nk),大量数据的话就jj了,我们没有必要维护窗口中的所有值,我们可以在队列中只保留可能成为窗口中的最大元素,去掉那些不可能成为窗口的最大元素。
如果新进来的数字大于滑动窗口的末尾元素,那么末尾元素就不可能再成为窗口中最大的元素了,因为这个大的数字是后进来的,一定会比之前先进入窗口的小的数字要晚离开窗口,因此我们就可以将滑动窗口中比其小的数字弹出队列,于是队列中的元素就会维持从队头到队尾单调递减,这就是单调递减队列。
如图所示:
对于队列中的元素来说:
- 在队列内自己左边的数就是数组中左边第一个比自己大的元素。
- 当被弹出时,遇到的就是数组中右边第一个比自己大的元素 ,只要元素还在队列中,就意味着暂时还没有数组中找到自己右侧比自己大的元素。
- 队头到队尾单调递减,队首元素为队列最大值。
动图演示:
实现细节:
- 因为过程中同时
弹出队首和队尾
,所以用双端队列
,队列中存的是下标
,为什么存下标呢?下标容易改变滑动窗口的范围,也可以对数组中的值操作 先检查合法性
:队头的下标距离超过了K,就出队单调性维护
:如果 nums[i] 大于或等于队尾元素下标所对应的值,则当前队尾再也不可能充当某个滑动窗口的最大值了,故需要队尾出队,始终保持队中元素从队头到队尾单调递减。队头的弹出
,由于单调队列中可能只有一个值(最大),因此不能通过队列长度和窗口长度比较来弹出,而需要判断当前队头元素是否等于当前窗口首位元素来决定是否弹出
代码如下:
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> q; //双端队列
vector<int> res; //返回答案数组
int n = nums.size();
for(int i = 0; i < n; ++i)
{
if(!q.empty() && i-k+1 > q.front())
{
q.pop_front(); //判断队头是否在滑动窗口内
}
while(!q.empty() && nums[i] >= nums[q.back()]){
q.pop_back(); //当要入队的元素大于队头时,队头出
}
q.push_back(i);//将元素放入队列
if(i >= k - 1) {
res.push_back(nums[q.front()]); //滑动窗口的元素达到了k个,才可以将其加入答案数组中
}
}
return res;
}
};
复杂度:
- 时间复杂度:每个元素最多遍历一遍,为O(N)
- 空间复杂度:队列中最多存K个数,O(k)
这题主要考察的是单调栈和单调队列的知识点。