题目大意:
给你n(<=10^6)个数和m,依次求出每个区间[i,i+m]的最大值与最小值。
题解:
n太大了,此题即使使用O(nlogm)的算法(比如线段树,RMQ等)的算法也有可能超时。考虑O(n)算法。
下面以求最大值为例:
设f[i]表示区间[i-m+1,i]的最大值,则有f[i]=max(a[j]),其中j满足b[i]<=j<i,其中b[i]=i-m+1;可见b[i]单调增加,满足单调队列条件。维护一个单调递减序列即可。
求最小值,则维护一个单调递增序列即可。
#include<iostream>
#include<cstdio>
#define inf (1<<30)
using namespace std;
int a[1000101],mina[1000101],maxa[1000101],q[1000101],tail,head;
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
head=1;tail=1;q[1]=1;
for(int i=1;i<=n;i++)
{
while(head<=tail&&q[head]<i-m+1)head++;
while(head<=tail&&a[i]>=a[q[tail]])tail--;
tail++;q[tail]=i;
maxa[i]=a[q[head]];
}
head=1;tail=1;q[1]=1;
for(int i=1;i<=n;i++)
{
while(head<=tail&&q[head]<i-m+1)head++;
while(head<=tail&&a[i]<=a[q[tail]])tail--;
tail++;q[tail]=i;
mina[i]=a[q[head]];
}
for(int i=m;i<=n-1;i++)
printf("%d ",mina[i]);
printf("%d\n",mina[n]);
for(int i=m;i<=n-1;i++)
printf("%d ",maxa[i]);
printf("%d\n",maxa[n]);
}
return 0;
}