求助分块WA最后一个点
查看原帖
求助分块WA最后一个点
285617
黑影洞人楼主2022/9/12 00:18
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
#include<cstring>
#define N 100010
using namespace std;
int n,m,a[N],s[N];
int ans[3000][3000],left[3000],right[3000],tot,block,pos[N];
int tmp[N];
vector<int>v[N];
void build(){
	n++;
	int mkx=0;
	block=sqrt(n/3);
	block=(int)pow(1.0*n,0.3333);
	tot=n/block;
	if(n%block)tot++;
	v[n].push_back(1);
	pos[1]=1;
	for(int i=2;i<=n;i++){
		scanf("%d",&a[i]);
		a[i]+=a[i-1];
		v[a[i]+n].push_back(i);
		pos[i]=(i-1)/block+1;
		mkx=max(mkx,a[i]+n);
	}
	for(int i=1;i<=n;i++)a[i]+=n;
	for(int i=1;i<=tot;i++){
		left[i]=(i-1)*block+1;
		right[i]=i*block;
	} 
	for(int i=1;i<=mkx;i++)tmp[i]=-1;
	right[tot]=n;
	for(int i=1;i<=tot;i++){
		for(int j=i;j<=tot;j++){
			int res=0;
			for(int k=left[pos[i]];k<=right[pos[j]];k++){
				if(tmp[a[k]]==-1)tmp[a[k]]=k;
				res=max(res,k-tmp[a[k]]);
				ans[pos[i]][pos[j]]=res;
			}
			for(int k=left[pos[i]];k<=right[pos[j]];k++)tmp[a[k]]=-1;
		}
	}
}
int query(int l,int r){
	int res=ans[pos[l]+1][pos[r]-1];r++;
	if(pos[l]==pos[r]){
		for(int i=l;i<=r;i++){
			vector<int>::iterator it;
			it=lower_bound(v[a[i]].begin(),v[a[i]].end(),r);
			it--;
			res=max(res,(*it)-i);
		}
	}else{
		res=ans[pos[l]+1][pos[r]-1];
		for(int i=l;i<=right[pos[l]];i++){
			vector<int>::iterator it;
			it=upper_bound(v[a[i]].begin(),v[a[i]].end(),r);
			it--;
			res=max(res,(*it)-i);
		}
		for(int i=left[pos[r]];i<=r;i++){
			vector<int>::iterator it;
			it=upper_bound(v[a[i]].begin(),v[a[i]].end(),l-1);
			res=max(res,i-(*it));
		}
	}
	return res;
}
signed main(){
	scanf("%d%d",&n,&m);
	build();
	while(m--){
		int l,r;
		scanf("%d%d",&l,&r);
		printf("%d\n",query(l,r)); 
	} 
	return 0;
}



2022/9/12 00:18
加载中...