数列分块入门5(区间开方)

原题:http://www.caioj.cn/problem.php?id=1249

题解:显然区间开方比较难做,但很容易发现2^32的数开4-5次方就会到1,考虑分块暴力,如果这个快全部是1的化就不用再维护,最坏的情况是1-n开4-5次方,可以接受。这道题加点小优化就可以特别快。

#include<bits/stdc++.h> 
#define N 50005
#define M 505
#define reg register 
using namespace std;
int n,m,pos[N],a[N],sum[N];
bool v[M];
inline int rd(){
	int x=0;int f=1;char s=getchar();
	while(s>'9' || s<'0') f=(f=='-'?-1:f),s=getchar();
	while(s>='0' && s<='9') x=(x<<1)+(x<<3)+s-'0',s=getchar();
	return x*f; 
}
inline void reset(int x){
	bool flag=1;sum[x]=0;
	for(reg int i=(x-1)*m+1;i<=x*m;i++) {
		a[i]=sqrt(a[i]);sum[x]+=a[i];if(a[i]>1) flag=0;
	}
	v[x]=flag;
}
inline void change(int l,int r){
	if(!v[pos[l]]){
		for(reg int i=l,flag;i<=min(r,pos[l]*m);i++){
			sum[pos[l]]-=a[i];a[i]=sqrt(a[i]);sum[pos[i]]+=a[i];	
		}
	}
	if(pos[l]!= pos[r] && !v[pos[r]]){
		for(reg int i=(pos[r]-1)*m+1;i<=r;i++) {
			sum[pos[i]]-=a[i];a[i]=sqrt(a[i]);sum[pos[i]]+=a[i];
		}
	}
	for(reg int i=pos[l]+1;i<=pos[r]-1;i++) if(!v[i]) reset(i);
}
inline int query(int l,int r){
	int ans=0;
	for(reg int i=l;i<=min(r,pos[l]*m);i++) ans+=a[i];
	if(pos[l]!=pos[r])
		for(reg int i=(pos[r]-1)*m+1;i<=r;i++) ans+=a[i];
	for(reg int i=pos[l]+1;i<=pos[r]-1;i++) ans+=sum[i];
	return ans;
}
int  main(){	
//	freopen("a1.in","r",stdin);
//	freopen("a.out","w",stdout);
	n=rd();m=sqrt(n); 
	for(reg int i=1;i<=n;i++) a[i]=rd(),pos[i]=(i-1)/m+1,sum[pos[i]]+=a[i];
	for(reg int i=1,opt,l,r,c;i<=n;i++){
		opt=rd();l=rd();r=rd();c=rd();
		if(opt==0) change(l,r);
		else printf("%d\n",query(l,r));	
	}
	return 0;
}

 

posted @ 2019-01-10 14:19  Exception2017  阅读(185)  评论(0编辑  收藏  举报