题目描述:
给定一个数组 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
提示:
1 <= nums.length <= 10^5
-10^4 <= nums[i] <= 10^4
1 <= k <= nums.length
来源:力扣(LeetCode)
思路:
维持一个单调队列,即让队列中的元素单调递增(或递减)。
这个单调队列可以用deque来实现,比较方便。
比较复杂的是push 的动作,当n比队列尾的元素大时,就从队尾出队,直到里面的元素大于n为止。
pop时,当队列头部等于n时则正常出队,若不等,则表明该元素已经在push时出队了。所以直接结束pop。
返回最大值,比较好理解,直接返回队首元素就好。
之后是对,窗口的使用:
思路是,先装满窗口,然后返回最大值,接着移动窗口,这样循环直到窗户到数组末尾结束。
class D_queue {
private:
deque<int> que;
public:
void push(int n){
while(!que.empty() && que.back()<n){
que.pop_back();
}
que.push_back(n);
}
void pop(int n){
if(!que.empty() && que.front()==n)
que.pop_front();
}
int max(){
return que.front();
}
};
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
D_queue window;
vector<int> res;
for(int i=0; i<nums.size(); ++i)
{
if(i<k-1){ // 先填满窗口
window.push(nums[i]);
}
else{ // 窗口向前滑动
window.push(nums[i]);
res.push_back(window.max());
window.pop(nums[i-k+1]);
}
}
return res;
}
};