省选专练之数据结构POI2010Pilots

单调队列裸题

目的是维护两个单调队列

一个是到目前为止最大值

一个是到目前为止最小值

完毕

#include<bits/stdc++.h>
using namespace std;
typedef int INT;
#define int long long
const int N=5e6+100;
inline void read(int &x){
	x=0;
	char ch=getchar();
	int f=1;
	while(ch<'0'||ch>'9'){
		if(ch=='-'){
			f=-1;
		}
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		x=x*10+ch-'0';
		ch=getchar();
	}
	x*=f;
}
struct Node{
	int sum;
	int idx;
}Qmx[N],Qmn[N];//单调递增队列/单调递减队列 
int headmx=0;
int tailmx=1;
int headmn=0;
int tailmn=1;
int n,k;
int sum[N]={};
int ans=1;
INT main(){
	read(k);
	read(n);
	for(int i=1;i<=n;i++){
		read(sum[i]);
	}	
//	n++;
//	sum[n]=1e16;
	headmx++;
	Qmx[headmx].idx=1;
	Qmx[headmx].sum=sum[1];
	headmn++;
	Qmn[headmn].idx=1;
	Qmn[headmn].sum=sum[1];
	for(int i=2;i<=n;i++){
		while(headmn<=tailmn&&Qmn[tailmn].sum>sum[i])tailmn--;
		while(headmx<=tailmx&&Qmx[tailmx].sum<sum[i])tailmx--;
		Qmx[++tailmx].sum=sum[i];
		Qmx[tailmx].idx=i;
		Qmn[++tailmn].sum=sum[i];
		Qmn[tailmn].idx=i;
		int last;
		while(Qmx[headmx].sum-Qmn[headmn].sum>k){
			if(Qmx[headmx].idx<Qmn[headmn].idx){
				last=Qmx[headmx].idx+1;
				headmx++;
			}
			else{
				last=Qmn[headmn].idx+1;
				headmn++;
			}
		}
		ans=max(ans,i-last+1);
	}
	cout<<ans;
}

猜你喜欢

转载自blog.csdn.net/fcb_x/article/details/80962587