题解——loj6281 数列分块入门5 （分块）

分块

若块内最大值为0或1，则不用再开方

然后暴力修改

可以证明，如果开方后向下取整，则最多开方4次一个数就会变成0或1

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
using namespace std;
long long n,sz,num,belong[],a[],maxb[],sum[];
void calbe(int n){
    for(int i=;i<=n;i++)
        belong[i]=(i-)/sz+;
}
void reset(int x){
    maxb[x]=;
    sum[x]=;
    for(int i=(x-)*sz+;i<=min(sz*x,n);i++){
        maxb[x]=max(maxb[x],a[i]);
        sum[x]+=a[i];
        }
}
long long query(int l,int r){
    int xl=belong[l];
    int xr=belong[r];
    long long ans=;
    for(int i=l;i<=min(xl*sz,(long long)r);i++)
        ans+=a[i];
    if(xl!=xr){
        for(int i=(xr-)*sz+;i<=r;i++)
            ans+=a[i];
    }
    for(int i=xl+;i<=xr-;i++)
        ans+=sum[i];
    return ans;
}
void update(int l,int r){
    int xl=belong[l];
    int xr=belong[r];
    for(int i=l;i<=min(xl*sz,(long long)r);i++){
        a[i]=sqrt(a[i]);
        }
    reset(xl);
    if(xl!=xr){
        for(int i=(xr-)*sz+;i<=r;i++){
            a[i]=sqrt(a[i]);
            }
        reset(xr);
    }
    for(int i=xl+;i<=xr-;i++){
        if(maxb[i]<=)
            continue;
        for(int j=(i-)*sz+;j<=i*sz;j++)
            a[j]=sqrt(a[j]);
        reset(i);
    }
}
int main(){
    scanf("%lld",&n);
    sz=sqrt(n);
    num=n/sz;
    if(n%sz)
        num++;
    calbe(n);
    for(int i=;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=;i<=num;i++)
        reset(i);
    for(int i=;i<=n;i++){
        int opt,l,r,c;
        scanf("%d %d %d %d",&opt,&l,&r,&c);
        if(opt==)
            update(l,r);
        else
            printf("%lld\n",query(l,r));
    }
    return ;
}