题目
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组[2, 3, 4, 2, 6, 2, 5, 1]及滑动窗口的大小3,那么一共存在6个滑动窗口,它们的最大值分别为[4, 4, 6, 6, 6, 5]。
题解
- 方法一 暴力做法,遍历每一个窗口 时间复杂度 O(kn) 省略
- 方法二 使用双端队列的方法
当我们遍历数组时,比如 [2, 3, 4, 2, 6, 2, 5, 1] 中,遍历到 6 的时候
此时滑动窗口的范围是 [4, 2, 6]
但是 6 前面的 4 和 2 都比 6 小
而且 4 和 2 要在 6 的前面,所以后面的滑动窗口中不会再有这个位置的 4 和 2 了
—————————
所以我们可以使用一个双端队列来维护滑动窗口的最大值
这个队列存储的是原数组的下标
这个队列从队头到队尾 下标的值递增,下标对应数组里的值递减(后面会解释)
—————————
遍历数组,当遍历到 i 时:
步骤1 如果队头的下标值到达当前 i 的距离大于 k 则这个队头对应的下标已经不在滑动窗口中了,应当剔除
步骤2 循环判断 如果队列中队尾的下标对应的值小于 nums[i] 则表明后面的滑动窗口中这个下标对应的值已经不可能再作为最大值出现了,应当删除。
例如
遍历到 6 的时候,队列中下标对应的值为 4,2
2 小于 6,删除2,队列变为 4
4 小于 6,删除4,队列变为空
步骤3 将 nums[i] 的下标 i 插入队列尾部
————————
解释为什么“队列从队头到队尾 下标的值递增,下标对应数组里的值递减”
因为每次往队列中插入数据的时候,都要从队尾开始,把所有下标对应值小于插入值的都删掉,然后再插入新值在数组中的位置。
·因为每次新值都插在队尾,所以队头到队尾存的下标值是递增的
·因为每次新值插入队尾的时候,比新值大的数都已经删掉,所以每次插入的时候,新值都是队列中最小的,所以队列中存储的下标对应的值递减
·所以队首存的值永远都是滑动窗口中最大值所对应的下标
代码
vector<int> maxInWindows(vector<int>& nums, int k) {
vector<int> res;
deque<int> q;
for(int i = 0; i < nums.size(); i++){
if(!q.empty() && i >= q.front() + k) q.pop_front();
while(!q.empty() && nums[q.back()] < nums[i]) q.pop_back();
q.push_back(i);
if(i >= k - 1) res.push_back(nums[q.front()]);
}
return res;
}