题目描述:
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组[2, 3, 4, 2, 6, 2, 5, 1]及滑动窗口的大小3,那么一共存在6个滑动窗口,它们的最大值分别为[4, 4, 6, 6, 6, 5]。
注意:
- 数据保证k大于0,且k小于等于数组长度。
样例
输入:[2, 3, 4, 2, 6, 2, 5, 1] , k=3
输出: [4, 4, 6, 6, 6, 5]
分析:
观察样例可以发现,解的出现顺序是自左向右的。也就是说,随着窗口的右移,解要么不变,要么右移。考虑到解的单调性,这里维护一个单调队列来解决该问题。
1.当前解(滑动窗口最大值)是在窗口最左边,也就是在队头,我们右移窗口时,便要将队头元素出队来防止滑动窗口元素个数超过k。
2.新加入滑动窗口的元素值大于等于队头元素,为了维护队单调递减的性质,只能将队头元素出队。(后加入的元素呆的时间更长,所以新加入元素等于队头元素时依然出队)
3.新加入的元素小于队头元素,此时不能直接入队。比如上例的滑动窗口滑动到了6 2 5,到5时是小于6的,但大于2,此时队列不单调。所以需要比较新加入元素与队尾元素,若队尾元素较小,则出队。
考虑到以上三点,我们需要一个队列在队头队尾都能出队,所以采用deque来解决。
注意:这里为了维护滑动窗口的大小,我们入队的是元素的下标而非值,使得操作更加方便。
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;
}
};