1、应用场景
滑动窗口问题,给定一个数组长度为n,在一个长度为k的窗口找最值
-
什么是滑动窗口?
在示例中,我们从数组中第一个元素开始遍历,由于窗口的大小是3,因此当遍历到第三个元素时,窗口就形成了。之后,继续遍历元素时,为了保持窗口的大小为3,左侧元素就需要从窗口中剔除。这样使得窗口一直在向右移动,直到考察到最后一个元素结束,这就是所谓的滑动窗口。
此图借大佬图
2、实现
-
首先开一个数组q[N]表示队列,队列里存原数组 a[i] 的下标 i
-
hh = 0, tt = -1队首,队尾指针
如上图:我们想找出窗口的最小值
例如第五次,我们可以看出,-3 在 3 的右面,且因为我们要找最小值,所以只要有-3的存在,就没有我3的存在,3 就永远用不上,-1也同理。
因此我们可以使用一个队列存储所有还没有被移除的下标。在队列中,这些下标按照从小到大的顺序被存储,并且它们在数组q[N]中对应的值是严格单调递增的(从队首到队尾)。
我们如何判断什么时候队头元素应该出队?
当:i - k + 1 > q[hh]的时候出队
因为,窗口不断往右移动,我们要保持队列的长度始终保持是3
3、例题:154. 滑动窗口 - AcWing题库
AC代码:
#include<iostream>
using namespace std;
const int N = 1e6+10;
int q[N],a[N];
int n,k;
int main()
{
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++)
{
//为什么这里不用while是因为我们要挪动窗口的话,只挪动一位,所以不用while
if(i-k+1 > q[hh]) hh++; //保持窗口数长度为3超过就队头元素出队
//如果队列里有元素找出最小的放入队列中
while(hh <= tt && a[q[tt]] >= a[i]) tt--;
//这个不能写在下面的if的下面是因为,有可能这个此时要存的这个数,就是最小的
//例如:1 3 -1 这组窗口
q[++tt] = i;
//窗口长度达到k输出一下最小值,最小值在队首
if(i >= k - 1) printf("%d ",a[q[hh]]);
}
puts("");//换行
hh = 0, tt = -1;
//找最大值同理上述
for(int i=0;i<n;i++)
{
if(i-k+1 > q[hh]) hh++;
while(hh <= tt && a[q[tt]] <= a[i]) tt--;
q[++tt] = i;
if(i >= k - 1) printf("%d ",a[q[hh]]);
}
puts("");
return 0;
}