分块RE30分求助
查看原帖
分块RE30分求助
418419
ko_no_lzx_da楼主2022/3/14 13:36
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;
long long L[100000],R[100000];
long long a[100000000],sum[1000000],n,m,len; 
int flag[100000],belong[100000];
void add(long long l,long long r){
	long long lk=belong[l],rk=belong[r];
	for(long long i=l;i<=min(r,R[lk]);i++){
		if(flag[lk])break;
		sum[lk]-=a[i];
		a[i]=sqrt(a[i]);
		sum[lk]+=a[i];
	}
	if(sum[lk]==R[lk]-L[lk]+1)flag[lk]=1;
	for(long long i=lk+1;i<=rk-1;i++){
		if(flag[i])continue;
		for(long long j=L[i];j<=R[i];j++){
			sum[i]-=a[j];
			a[j]=sqrt(a[j]);
			sum[i]+=a[j];
		}
		if(sum[i]==R[i]-L[i]+1)flag[i]=1;
	}
	if(lk!=rk){
		for(long long i=L[rk];i<=r;i++){
			if(flag[rk])break;
			sum[rk]-=a[i];
			a[i]=sqrt(a[i]);
			sum[rk]+=a[i];
		}
		if(sum[rk]==R[rk]-L[rk]+1)flag[rk]=1;
	}
	return;
}
long long visit(long long l,long long r){
	long long ans=0;
	long long lk=belong[l],rk=belong[r];
	for(long long i=l;i<=min(r,R[lk]);i++){
		ans+=a[i];
	}
	for(long long i=lk+1;i<=rk-1;i++){
		ans+=sum[i];
	}
	if(lk!=rk){
		for(long long i=L[rk];i<=r;i++){
			ans+=a[i];
		}
	}
	return ans;
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin >>n;
	for(long long i=1;i<=n;i++){
		cin >>a[i];
	}
	len=sqrt(n);
	for(long long i=1;i<=len;i++){
		L[i]=(i-1)*n+1;
		R[i]=i*n;
	}
	if(R[len]!=n){
		len++;
		L[len]=R[len-1]+1;
		R[len]=n;
	}
	for(long long i=1;i<=len;i++){
		for(long long j=L[i];j<=R[i];j++){
			belong[j]=i;
			sum[i]+=a[j];
		}
	}
	cin >>m;
	for(long long i=1;i<=m;i++){
		long long k,l,r;
		cin >>k>>l>>r;
		if(l>r)swap(l,r);
		if(k==0){
			add(l,r);
		}else{
			cout <<visit(l,r)<<endl;
		}
	}
	return 0;
}


2022/3/14 13:36
加载中...