✍个人博客:https://blog.csdn.net/Newin2020?spm=1011.2415.3001.5343
📚专栏地址:剑指offer系列题解
📝原题地址:题目地址
📣专栏定位:为找工作的小伙伴整理常考算法题解,祝大家都能成功上岸!
❤️如果有收获的话,欢迎点赞👍收藏📁,您的支持就是我创作的最大动力💪
题目描述
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组 [2,3,4,2,6,2,5,1] 及滑动窗口的大小 3,那么一共存在 6 个滑动窗口,它们的最大值分别为 [4,4,6,6,6,5]。
注意:
- 数据保证 k 大于 0,且 k 小于等于数组长度。
数据范围
数组长度 [1,1000]。
样例
输入:[2, 3, 4, 2, 6, 2, 5, 1] , k=3 输出: [4, 4, 6, 6, 6, 5]
方法一:单调队列 O(n)
我们可以用一个单调队列来存储元素的下标,队头存储的是当前区间中最大值在数组中的下标,队头到队尾的元素呈递减趋势,思路如下:
- 从前往后遍历每个元素,如果单调队列的队头已经不在当前区间内,则弹出该元素的下标。
- 如果队尾下标所属元素小于等于当前遍历到的元素,就弹出该元素下标。
- 将当前元素的下标加入队尾,如果当前遍历的范围已经大于等于区间大小,则记录最大值。
我们拿题目的样例进行举例,假设给定数组 [2, 3, 4, 2, 6, 2, 5, 1]
,且 k = 3
。
第一轮: 由于此时窗口并没有超载,所以不用删除队列中的元素,直接将第一个元素下标 0
加入加入队列中。同时,由于当前窗口内数量还未达到 3
,故不用加入答案数组 ans
中。
第二轮: 同样,窗口还未超载,队头元素下标仍然在可控范围内,但是可以发现第二个元素 3
已经比队列中队尾存的下标所指元素 nums[0] = 2
要大,故将队列中该元素踢出,并加入当前所指元素的下标 1
。同时,由于当前窗口内数量还未达到 3
,故不用加入答案数组 ans
中。
第三轮: 同第二轮,nums[2] = 4 > nums[1] = 3
,故将下标 1
踢出队列,加入下标 2
。同时,由于当前窗口内数量已经达到 3
,故将队头元素 nums[2] = 4
加入答案数组 ans
中。
第四轮: 由于队头所指元素 nums[2]
并未超出当前窗口,故不用删除元素。同时,第四个元素 nums[3] = 2
小于队尾元素 nums[2] = 4
,故直接加到队尾,并且将队头元素 nums[2] = 4
加入答案数组 ans
中。
第五轮: 由于队头所指元素 nums[2]
并未超出当前窗口,故不用删除元素。但是第五个元素 nums[4] = 6
大于队中两个元素 nums[2] = 4
和 nums[3] = 2
,故删除队中两个元素并加入第五个元素下标 4
,并且将队头元素 nums[4] = 6
加入答案数组 ans
中。
第六轮: 由于队头所指元素 nums[4]
并未超出当前窗口,故不用删除元素。同时,第六个元素 nums[5] = 2
小于队尾元素 nums[4] = 6
,故直接加到队尾,并且将队头元素 nums[4] = 6
加入答案数组 ans
中。
第七轮: 由于队头所指元素 nums[4]
并未超出当前窗口,故不用删除元素。同时,第七个元素 nums[6] = 5
大于队尾元素 nums[5] = 2
,故删除队尾元素,但 nums[6] = 5
小于 nums[4]
,故直接加到队尾,并且将队头元素 nums[4] = 6
加入答案数组 ans
中。
第八轮: 此时队头元素 4
已经在当期窗口 5 ~ 7
外,故需要删除队头元素。同时第八个元素小于当前队尾元素 nums[6]
,故直接加到队尾,并且将队头元素 nums[6] = 5
加入答案数组 ans
中。
至此,遍历结束,最后返回答案数组 ans
即可。
class Solution {
public:
vector<int> maxInWindows(vector<int>& nums, int k) {
vector<int> ans;
deque<int> q;
for (int i = 0; i < nums.size(); i++)
{
while (q.size() && q.front() <= i - k) q.pop_front();
while (q.size() && nums[q.back()] <= nums[i]) q.pop_back();
q.push_back(i);
if (i - k >= -1) ans.push_back(nums[q.front()]);
}
return ans;
}
};
欢迎大家在评论区交流~