线段树 点更新 模板

#include <stdio.h>
#define N 100005
int a[N],f[4*N];
int n;
void build(int l,int r,int now){
    if(l==r){
        f[now]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
    f[now]=f[now<<1]+f[now<<1|1];
}
int search(int l,int r,int ll,int rr,int now){
    if(l==ll&&r==rr){
        return f[now];
    }
    int mid=(l+r)>>1;
    if(ll>mid) return search(mid+1,r,ll,rr,now<<1|1);
    else if(rr<=mid) return search(l,mid,ll,rr,now<<1);
    else return search(mid+1,r,ll,rr,now<<1|1)+search(l,mid,ll,rr,now<<1);
}
void mix(int l,int r,int p,int t,int now){
    if(l==r){
        f[now]=t;
    }
    int mid=(l+r)>>1;
    if(p<=mid) mix(1,mid,p,t,now<<1);
    else mix(mid+1,r,p,t,now<<1|1);
    f[now]=f[now<<1]+f[now<<1|1];
}
int main(){
    int l,r;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    build(1,n,1);
    int m,op,pos,t;
    scanf("%d",&m);
    while(m--){
        for(int i=0;i<25;i++){
            printf("%d ",f[i]);
        }
        if(op==1){
            scanf("%d%d",&l,&r);
            printf("%d\n",search(1,n,l,r,1));
        }
        else{
            scanf("%d%d",&pos,&t);
            mix(1,n,pos,t,1);
        }
    }
}

猜你喜欢

转载自blog.csdn.net/MallowFlower/article/details/81390055
今日推荐