给出一个可能包含重复的整数数组,和一个大小为 k 的滑动窗口, 从左到右在数组中滑动这个窗口,找到数组中每个窗口内的最大值。
样例
给出数组 [1,2,7,7,8]
, 滑动窗口大小为 k = 3
. 返回 [7,7,8]
.
解释:
最开始,窗口的状态如下:
[|1, 2 ,7| ,7 , 8]
, 最大值为 7
;
然后窗口向右移动一位:
[1, |2, 7, 7|, 8]
, 最大值为 7
;
最后窗口再向右移动一位:
[1, 2, |7, 7, 8|]
, 最大值为 8
.
挑战
O(n)时间,O(k)的额外空间
解题思路:
维持一个双端队列qmax,里面存放着nums中元素的下标,这些下标对应的元素在qmax中严格递减,所以qmax的队头对应的元素最大,队尾对应的元素最小。
left位于滑动窗口的左端,如果qmax中的元素过期(即元素位于left的左边)则弹出,qmax存放下标的好处就是可以和left直接比较判断是否过期
class Solution {
public:
vector<int> maxSlidingWindow(vector<int> &nums, int k) {
if(nums.empty())
return vector<int>();
deque<int> qmax;
vector<int> res;
int size=nums.size();
for(int i=0;i<k;++i){
while(!qmax.empty()&&nums[qmax.back()]<=nums[i]) //维持元素严格递减
qmax.pop_back();
qmax.push_back(i);
}
res.push_back(nums[qmax.front()]); //放入第一个最大值
for(int i=k,left=1;i<size;++i,++left){
while(!qmax.empty()&&nums[qmax.back()]<=nums[i]) //继续维持
qmax.pop_back();
qmax.push_back(i);
if(left>qmax.front()) //判断是否过期,滑动窗口一直向左滑动的同时可能最大值已经处于窗口之外了(即过期)
qmax.pop_front();
res.push_back(nums[qmax.front()]);
}
return res;
}
};