bzoj 3038 线段树

维护一个标记,当值为0或1时便不用修改

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++) 
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=1e5+10;
int n,m;ll a[N];
struct Node{int l,r,tag;ll sum;}tr[N<<2];
void build(int x,int l,int r)
{
    tr[x].l=l; tr[x].r=r; tr[x].sum=tr[x].tag=0;
    if(l==r) {tr[x].sum=a[l]; if(a[l]==1||a[l]==0) tr[x].tag=1;return ;}
    int mid=(l+r)>>1;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    tr[x].sum=tr[x<<1].sum+tr[x<<1|1].sum;
    tr[x].tag=tr[x<<1].tag&tr[x<<1|1].tag;

}
void update(int x,int ql,int qr)
{
    int l=tr[x].l,r=tr[x].r,mid=(l+r)>>1;
    if(tr[x].tag) return ;
    if(l==r) {
        tr[x].sum=floor(sqrt(tr[x].sum));
        if(tr[x].sum==1||tr[x].sum==0) tr[x].tag=1;
        return ;
    }
    if(qr<=mid) update(x<<1,ql,qr);
    else if(ql>mid) update(x<<1|1,ql,qr);
    else {
        update(x<<1,ql,mid);
        update(x<<1|1,mid+1,qr);
    }
    tr[x].sum=tr[x<<1].sum+tr[x<<1|1].sum;
    tr[x].tag=tr[x<<1].tag&tr[x<<1|1].tag;
}  
ll query(int x,int ql,int qr)
{
    int l=tr[x].l,r=tr[x].r,mid=(l+r)>>1;   
    if(ql==l&&r==qr) {
        return tr[x].sum;
    }
    if(qr<=mid) return query(x<<1,ql,qr);
    else if(ql>mid) return query(x<<1|1,ql,qr);
    else return (query(x<<1,ql,mid)+query(x<<1|1,mid+1,qr)); 
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n) scanf("%lld",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    for(int k,l,r,i=1;i<=m;i++) {
        scanf("%d%d%d",&k,&l,&r);
        if(l>r) swap(l,r);
        if(k==0) update(1,l,r);
        else printf("%lld\n",query(1,l,r));
    }
    return 0;
}