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