题目
给定一个数组 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
思路
直接模拟复杂度O(kn)
利用双向队列实现O(n)
这里就注意一点,为什么用数组下标而不是数值
代码
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> q;
int len = nums.size();
vector<int> ret;
if(len >=k && k >= 1){
//还没有完全形成滑动窗口的情况下
for(int i = 0;i < k;i++){
//只要输入进来的数比队列已有的最后一个大,就应该弹出队尾的,因为这个数不可靠再成为大数
while(q.empty() != true && nums[i] >= nums[q.back()]){
q.pop_back();
}
q.push_back(i);
}
for(int i = k;i < len;i++){
ret.push_back(nums[q.front()]);
while(q.empty() != true && nums[i] >= nums[q.back()]){
q.pop_back();
}
//如果下标之差大于等于滑动窗口大小,这个数字已经滑出
if(q.empty() != true && q.front() <= i-k){
q.pop_front();
}
q.push_back(i);
}
ret.push_back(nums[q.front()]);
}
return ret;
}
};