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;
}

转载于:https://www.cnblogs.com/sdzwyq/p/9837687.html