给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 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 ≤ 输入数组的大小,且输入数组不为空。
我们可以用双向队列deque来做这道题
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> res;
deque<int> q;
for(int i=0;i<nums.size();++i)
{
//如果队列中已经有三个元素了,就出去一个
if(!q.empty()&&q.front()==i-k) q.pop_front();
//保证队列最头部的元素一定是最大的
while(!q.empty()&&(nums[i]>nums[q.back()])) q.pop_back();
q.push_back(i);
//因为头部为最大,所以头部进数组就可以了
if(i>=k-1) res.push_back(nums[q.front()]);
}
return res;
}
};
还有一种解法:有有限队列,直接上代码,但是leetcode测试过了,提交超时了,时间复杂度貌似真的高了。。。。。
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
if(nums.empty())
return {};
vector<int> res;
for(int i=0;i<nums.size()-k+1;i++)
{
priority_queue<int> q;
for(int t=0;t<k;t++)
q.push(nums[i+t]);
res.push_back(q.top());
}
return res;
}
};
这道题和 leetcode703. 数据流中的第K大元素 解法很像