滑动窗口,就是有一个区间,当区间长度大于k时,就pop该区间的首元素,这满足先进先出的条件,那么我们可以用一个队列来维护这个区间
#include<iostream>
using namespace std;
const int N=1000010;
int a[N],queue[N];
int n,k;
int main(void)
{
scanf("%d %d",&n,&k);
for(int i=0;i<n;i++) scanf("%d",&a[i]);
int hh=0,tt=-1;//hh为队头,tt为队尾
for(int i=0;i<n;i++)
{
if(hh<=tt&&i-k+1>queue[hh])//判断是否需要滑动窗口,pop队头
{
hh++;
}
while(hh<=tt&&a[i]<a[queue[tt]])//判断队列末尾是否满足单调性
{
tt--;
}
queue[++tt]=i;
if(i-k+1>=0) printf("%d ",a[queue[hh]]);//如果滑动窗口满足k,可进行输出
}
printf("\n");
hh=0,tt=-1;
for(int i=0;i<n;i++)
{
if(hh<=tt&&i-k+1>queue[hh])
{
hh++;
}
while(hh<=tt&&a[i]>a[queue[tt]])
{
tt--;
}
queue[++tt]=i;
if(i-k+1>=0) printf("%d ",a[queue[hh]]);
}
return 0;
}
变式:
#include<iostream>
using namespace std;
const int N = 5000010;
int a[N], queue[N];
int n, m;
int main(void)
{
scanf("%d %d", &n, &m);
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 - m> queue[hh])hh++;
while (hh <= tt && a[i] < a[queue[tt]])tt--;
if (i>= 1)printf("%d\n", a[queue[hh]]);
else printf("0\n");
queue[++tt] = i;
}
return 0;
}