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