题目
解释
- 考虑单调栈和队列的做法,首先考虑暴力做法,然后将暴力做法中重复无用的元素删除
- 暴力做法是O(NK),即每K个元素为一组遍历一次,一共有N-K+1组
- 如果本题用STL做而不是数组模拟,则速度会慢很多
- 无论是进行弹出还是插入操作,都要保证队列里始终有内容,即hh<=tt;
- 对于求最小值,弹出的规则是,如果当前窗口的元素为a b c,当c<=b,a时,则可以将b和a直接弹出,然后c插入到队头。在这之后队头永远是最小的元素
演示视频
代码段
#include<iostream>
using namespace std;
const int N=1e6+10;
int q[N],a[N],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++)
{
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) printf("%d ",a[q[hh]]);
//if判断来确保滑动窗口的初始位置里有k个元素
}
cout<<endl;
//寻找最大值同理
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) printf("%d ",a[q[hh]]);
}
cout<<endl;
return 0;
}