题目大意:给定n个数,有一个长度为k的窗口从左滑至右,依次输出窗口内的最大值与最小值。
思路:显然,这是单调队列的模板题。这里说一下单调队列:这是一种具有单调性的队列,以求最大值为例,将出现的数据按时间(即从左向右)放入队列中。当一个元素被从尾部放入队列中时,考察已在队列中的末尾元素,如果比要插入的元素值小,则一直删至一个比它大的元素值或队列为空为止。这样队列内部就是单调递减的。这样做的合理性在于,在这个窗口中,这些被删去的元素都不可能再被用到,因为现在插入的元素值已经占据了将来可能的最大值的位置。如果窗口长度过长(即长于规定长度),就将队列首位元素弹出,因为此时窗口已不再包含该元素。这样做之所以快速高效,是因为它保留了可能成为最大值的值,也因此省去了重复查找的时间。
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<deque>
using namespace std;
struct node
{
int num,id;
node(int num=0,int id=0):num(num),id(id){}
};
const int maxn=1000005;
int n,k,a[maxn],minnum[maxn],maxnum[maxn];
void init()
{
scanf("%d%d",&n,&k);
for (int i=1;i<=n;++i)
scanf("%d",&a[i]);
}
deque<node> qmax,qmin;
void work()
{
qmax.push_back(node(a[1],1));
qmin.push_back(node(a[1],1));
int minp=0,maxp=0;
for (int i=2;i<=k;++i)
{
while (!qmax.empty() && i-qmax.front().id>=k)
qmax.pop_front();
while (!qmax.empty() && a[i]>=qmax.back().num)
qmax.pop_back();
qmax.push_back(node(a[i],i));
while (!qmin.empty() && i-qmin.front().id>=k)
qmin.pop_front();
while (!qmin.empty() && a[i]<=qmin.back().num)
qmin.pop_back();
qmin.push_back(node(a[i],i));
}
minnum[++minp]=qmin.front().num;
maxnum[++maxp]=qmax.front().num;
for (int i=k+1;i<=n;++i)
{
while (!qmax.empty() && i-qmax.front().id>=k)
qmax.pop_front();
while (!qmax.empty() && a[i]>=qmax.back().num)
qmax.pop_back();
qmax.push_back(node(a[i],i));
while (!qmin.empty() && i-qmin.front().id>=k)
qmin.pop_front();
while (!qmin.empty() && a[i]<=qmin.back().num)
qmin.pop_back();
qmin.push_back(node(a[i],i));
minnum[++minp]=qmin.front().num;
maxnum[++maxp]=qmax.front().num;
}
for (int i=1;i<=minp;++i)
printf("%d ",minnum[i]);
printf("\n");
for (int i=1;i<=maxp;++i)
printf("%d ",maxnum[i]);
}
int main()
{
init();
work();
return 0;
}