滑动窗口
题目
给定一个大小为 n≤10^6 的数组。有一个大小为 k 的滑动窗口,它从数组的最左
边移动到最右边。你只能在窗口中看到 k 个数字。每次滑动窗口向右移动一个位
置。以下是一个例子:
该数组为 [1 3 -1 -3 5 3 6 7],k 为 3。
你的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。
输入样例:
8 3
1 3 -1 -3 5 3 6 7
输出样例:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
代码
#include<bits/stdc++.h>
using namespace std;
int main(){
int nums[10];
int n;
int k=3;
scanf("%d", &n);
for(int i=0;i<n;++i){
scanf("%d", &nums[i]);
}
vector<int> q(n+1, 0);
int hh=0, tt=-1; // hh和tt存放的是队列的首尾指针
for(int i=0;i<n;++i){
if(hh<=tt&&i+1-q[hh]>k) ++hh;
while(hh<=tt&&nums[q[tt]]>=nums[i]) --tt;
q[++tt]=i; // 队列里存放的是数组的下标(指针)
if(i+1-k>=0){
printf("%d ", nums[q[hh]]);
}
}
return 0;
}
知识点补充
单调队列解决:滑动窗口的问题。
单调栈解决:找出当前元素右边的第一个比它大的元素
单调递增栈: 栈顶最大
单调递减栈: 栈顶最小
单调递增队列: 队首最小
单调递减队列:队首最大
方向: 首------->尾(底-------->尾)
单调队列的操作:
1, 单调队列用C++ STL deque实现
2, 队尾插入元素时, 当前元素和队尾元素是否满足单调性。可能发生出队列操作。
3,当队首元素的位置不在合法区间内,那么队首元素弹出。
4, 代码2可以形成一个模版,记忆即可。