题目链接
题目描述
给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。
示例:
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:
滑动窗口的位置 最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
提示:
你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。
解题思路
利用双端队列,首先将前k个元素放入双端队列中
这里的一个重要操作是将要在队尾插入的最大元素与原本队尾的元素进行比较,如果即将插入的元素大于原本的队尾元素,就将原队尾出栈,这样可以保证在deque中元素一定是从大到小排列的
然后再从队头向后(队列中从大到小)判断,如果当前元素的下标不在窗口中,那么一直出栈,直到找到的第一个在窗口中的元素就是当前窗口的最大元素
题解
class Solution {
public:
vector<int> v;
deque<int> q;
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
if(k==0) return {};
for(int i=0;i<k;i++){
while(!q.empty()&&nums[i]>=nums[q.back()]) q.pop_back();
q.push_back(i);
}
v.push_back(nums[q.front()]);
for(int i=k;i<nums.size();i++){
while(!q.empty()&&nums[i]>=nums[q.back()]){
q.pop_back();
}
q.push_back(i);
while(q.front()<=i-k) q.pop_front();
v.push_back(nums[q.front()]);
}
return v;
}
};