BZOJ3211花神游历各国

BZOJ3211花神游历各国

BZOJ
luogu
分块
记一个all表示该块是否全部<=1,如果all不为真就暴力修改
因为一个数被开根的次数不多,即使\(10^{12}\)只要开根6次也会变成1,所以复杂度是可以证明的
注意BZOJ数据含0

#define ll long long
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int re(){
	int x=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*w;
}
int n,m,B,a[N],b[N];
bool all[N];
ll sum[N];
bool check(int x){
	for(int i=(x-1)*B+1;i<=min(n,x*B);i++)
		if(a[i]!=1&&a[i])return 0;
	return 1;
}
void modify(int l,int r){
	if(!all[b[l]]){
		for(int i=l;i<=min(r,b[l]*B);i++){
			if(a[i]==1||!a[i])continue;
			int sq=sqrt(a[i]);
			sum[b[i]]-=a[i]-sq;a[i]=sq;
		}
		if(check(b[l]))all[b[l]]=1;
	}
	if(b[l]!=b[r]){
		if(!all[b[r]]){
			for(int i=(b[r]-1)*B+1;i<=r;i++){
				if(a[i]==1||!a[i])continue;
				int sq=sqrt(a[i]);
				sum[b[i]]-=a[i]-sq;a[i]=sq;
			}
			if(check(b[r]))all[b[r]]=1;
		}
	}
	for(int i=b[l]+1;i<b[r];i++){
		if(all[i])continue;
		for(int j=(i-1)*B+1;j<=i*B;j++){
			if(a[j]==1||!a[j])continue;
			int sq=sqrt(a[j]);
			sum[i]-=a[j]-sq;a[j]=sq;
		}
		if(check(i))all[i]=1;
	}
}
ll query(int l,int r){
	ll res=0;
	for(int i=l;i<=min(r,b[l]*B);i++)res+=a[i];
	if(b[l]!=b[r]){
		for(int i=(b[r]-1)*B+1;i<=r;i++)res+=a[i];
	}
	for(int i=b[l]+1;i<b[r];i++)res+=sum[i];
	return res;
}
int main(){
	n=re();B=sqrt(n);
	for(int i=1;i<=n;i++){
		a[i]=re(),b[i]=(i-1)/B+1;
		sum[b[i]]+=a[i];
	}
	m=re();
	while(m--){
		int k=re(),l=re(),r=re();
		if(k==2)modify(l,r);
		else printf("%lld\n",query(l,r));
	}
	return 0;
}