这节课我们来讲滑动窗口。滑动窗口实质上是对于单调队列的应用,本质上其实和之前的单调栈是同一类型的替身。
我们来看这道题ACwing 滑动窗口
我们针对这道题开始分析。
1(分析题意).简单来看题意其实就是给一串数组(长度为n),我们模拟一个在刚刚数组上的窗口(长度为k),求窗口在从前向后滑动的过程当中,窗口中的最大值和最小值。
2.(暴力求解).从前向后遍历n-k个数字O(n),得到n-k个窗口,在每一个窗口中求得窗口中的最大值和最小值O(k),最后的时间复杂度是O(n*k)。对于一些样例得分的比赛,这样做对于新手来说基本够了,但如果是acm赛事的比赛,那肯定大部分样例是过不去的。
3.(算法优化).我们寻找窗口移动时的规律,可以将最大值和最小值分开来看,先只找最小值,发现有一些数字总是不会被输出的,他们满足这个性质:在他们同一个窗口中,有数字比他们小。我们利用这个性质,可以生成一个递增单调队列来满足这个性质,即从前往后开始维护一个递增的队列,如果有数字比之前的数字大,那么就将之前的数字弹出队列。说到这里,是不是觉得和单调栈很像,原理相同,但这个单调队列的维护方式与单调栈有很大的不同。
int q[N], a[N];
int n, k;
scanf("%d%d",&n,&k);
for(int i = 0; i < n; i ++ ) scanf("%d",&a[i]);
int hh = 0, tt = -1;
for(int i = 0; i < n; i ++ ){
if(hh <= tt && i - k + 1 > q[hh]) hh++;
while(hh <= tt && a[q[tt]] >= a[i]) tt--;
q[++tt] = i;
if(i - k + 1 >= 0) printf("%d ",a[q[hh]]);
}//单调队列
int stk[N], tt;
int n;
cin >> n;
while(n --){
int x;
cin >> x;
while(tt && stk[tt] >= x) tt --;
if(!tt) printf("-1 ");
else printf("%d ",stk[tt]);
stk[++tt] = x; //单调栈
}
同样是从小到大,单调栈和单调队列的维护方式不太一样。单调栈是直接创造一个数组来模拟栈,如果栈顶元素大于x,那么就弹出栈顶元素,x作为栈顶元素,这个比较好理解。而单调队列是先创建一个数组a[N]储存原数组,之后创造数组q[N]来维护单调队列,设置了头指针hh,尾指针tt。
我们来理解一下每句话的含义
首先定义q[N],a[N],n(数组长度),k(窗口长度),之后在循环里面,if(hh <= tt && i - k + 1 > q[hh]) hh++; 表示如果在尾指针大于等于头指针的情况下,如果营造的窗口长度大于题中给定的窗口长度,那么hh++,也就是如果窗口大了,那么就往前挪一点。
while(hh <= tt && a[q[tt]] >= a[i]) tt--; 表示如果窗口中最后一个数小于新的数字,那么就将末尾的数弹出,重复此操作。
q[++tt] = i; 表示将新的数字存入窗口尾部
if(i - k + 1 >= 0) printf("%d ",a[q[hh]]); 如果满足窗口的长度,那么就开始输出
个人建议,如果想要深刻理解,最好自己手动模拟一下,如果只是想要应用,那么就只需要理解每句话的含义,之后背会就基本OK。