给定一个数组 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 ≤ 输入数组的大小,且输入数组不为空。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sliding-window-maximum
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> vecRes;
if(nums.size() == 0) return vecRes;
deque<int> de;
for(int i = 0; i < nums.size(); i++)
{
if(!de.empty() && de.front() == i - k)
de.pop_front();
while(!de.empty() && nums[de.back()] <= nums[i])
de.pop_back();
de.push_back(i);
if(i >= k-1)
vecRes.push_back(nums[de.front()]);
}
return vecRes;
}
};
使用双端队列,队列头存放窗口最大元素索引,当队列中的元素个数大于窗口数时,移除队列尾元素,每次遍历时,若队列尾端元素不大于当前元素,移除队尾元素,每次将当前元素放到队列尾部;若索引大于等于k-1,则将队首元素放到vector中。