这是一个很明显的单调队列问题,数据有100万,如果用其他的方法就很有可能超时。。。求最大值和最小值所以要有两个单调队列一个球最大值,一个球最小值。。
其实单调队列里面只有最优值和次优值有用(感觉其他值基本可以忽略),所以直接去值就可以了。。。
最大队列保证队列中各个元素大小单调递减(即,最大元素在对头,最小的在队尾)。这样便保证队首元素最大,而且更新的时候队首永远是当前最大。因此,这个队列需要在两头都可以进行删除,在队尾插入。
维护方法:在每次插入的时候,先判断队尾元素,如果不比待插入元素大就删除,不断删除队尾直到队尾元素大于待插入元素或者队空。删除的时候,判断队首,如果队首元素下标小于当前段左边界就删除,不断删除队首直到队首元素下标大于等于当前段左边界(注意:这时队列肯定不为空),队首元素就是当前段的最优解。
代码:
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int mm=1e6+8;
struct node{
int x,y;
}q[mm],w[mm];
int minn[mm],maxx[mm],ss[mm];
int main(){
int i,j,n,k,t;
int f1,f2,r1,r2;
while(~scanf("%d%d",&n,&k)){
if(k > n) k=n;
memset(q,0,sizeof(q));
memset(w,0,sizeof(w));
for(i=1;i<=n;i++){
scanf("%d",&ss[i]);
}
f1=f2=0;
r1=r2=0;
for(i=1;i<k;i++){
while(f1 <= r1 && q[r1].x < ss[i]) r1--;
q[++r1].x=ss[i];
q[r1].y=i;
while(f2 <= r2 && w[r2].x > ss[i]) r2--;
w[++r2].x=ss[i];
w[r2].y=i;
}
for(i=k;i<=n;i++){ //f1和f2保存了查找的起点,前面保存的数据已经无效。。
while(f1 <= r1 && q[r1].x < ss[i]) r1--;
q[++r1].x=ss[i];
q[r1].y=i;
while(f1 <= r1 && q[f1].y <= i-k) f1++;
maxx[i-k]=q[f1].x;
while(f2 <= r2 && w[r2].x > ss[i]) r2--;
w[++r2].x=ss[i];
w[r2].y=i;
while(f2 <= r2 && w[f2].y <= i-k) f2++;
minn[i-k]=w[f2].x;
//printf(" r1=%d,f1=%d; r2=%d,f2=%d\n",r1,f1,r2,f2);
}
for(i=0;i<n-k;i++) printf("%d ",minn[i]);
printf("%d\n",minn[n-k]);
for(i=0;i<n-k;i++) printf("%d ",maxx[i]);
printf("%d\n",maxx[n-k]);
}
return 0;
}