数列分块入门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;
}