【bzoj3038】 上帝造题的七分钟2

#include<cstdio>
#include<cmath>
#include<iostream>
using namespace std;
long long n,m,a[100001];
struct data{
    int l,r;long long sum;
    bool flag;
}tr[400001];
long long ask(int k,int s,int t){
    int l=tr[k].l,r=tr[k].r;
    if(l==s&&t==r)return tr[k].sum;
    int mid=(l+r)>>1;
    if(t<=mid)return ask(k<<1,s,t);
    else if(s>mid)return ask(k<<1|1,s,t);
    else return ask(k<<1,s,mid)+ask(k<<1|1,mid+1,t);
}
void build(int k,int s,int t){
    tr[k].l=s;tr[k].r=t;
    if(s==t){
        tr[k].sum=a[s];
        if(a[s]==1||!a[s])tr[k].flag=true;
        return;
    }
    int mid=(s+t)>>1;
    build(k<<1,s,mid);
    build(k<<1|1,mid+1,t);
    tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum;
    tr[k].flag=tr[k<<1].flag&tr[k<<1|1].flag;
}
void update(int k,int s,int t){
    if(tr[k].flag)return;
    int l=tr[k].l,r=tr[k].r;
    if(l==r){
        tr[k].sum=(long long)sqrt(tr[k].sum);
        if(tr[k].sum==1||!tr[k].sum)tr[k].flag=true;
        return;
    }
    int mid=(l+r)>>1;
    if(t<=mid)update(k<<1,s,t);
    else if(s>mid)update(k<<1|1,s,t);
    else{
        update(k<<1,s,mid);
        update(k<<1|1,mid+1,t);
    }
    tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum;
    tr[k].flag=tr[k<<1].flag&tr[k<<1|1].flag;
}
 
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int t,l,r;
        scanf("%d %d %d",&t,&l,&r);
        if(l>r)swap(l,r);
        if(!t)update(1,l,r);
        else printf("%lld\n",ask(1,l,r));
    }
    return 0;
}