首页 > 技术文章 > 线段树区间修改

zookkk 2018-11-28 20:35 原文

再补个线段树区间修改的板子

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=1e5+9;
struct node{
    int lazy,val,l,r;
}T[maxn*4];
void pushup(int rt){
    T[rt].val=T[rt<<1].val+T[rt<<1|1].val;
    return;
}
void pushdown(int rt){
    if(T[rt].lazy){
        T[rt<<1].lazy+=T[rt].lazy;
        T[rt<<1|1].lazy+=T[rt].lazy;
        T[rt<<1].val+=T[rt].lazy*(T[rt<<1].r-T[rt<<1].l+1);
        T[rt<<1|1].val+=T[rt].lazy*(T[rt<<1|1].r-T[rt<<1|1].l+1);
        T[rt].lazy=0;
    }
}
void build(int l,int r,int rt){
    T[rt].l=l;T[rt].r=r;
    if(l==r){
        cin>>T[rt].val;
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
void update(int L,int R,int l,int r,int rt,int val){
    if(l>=L&&r<=R){
        T[rt].val+=(r-l+1)*val;
        T[rt].lazy+=val;
        return;
    }
    pushdown(rt);
    int mid=(l+r)>>1;
    if(mid>=L)update(L,R,l,mid,rt<<1,val);
    if(mid<R)update(L,R,mid+1,r,rt<<1|1,val);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt){
    if(l>=L&&r<=R){
        return T[rt].val;
    }
    pushdown(rt);
    int mid=(r+l)>>1;
    int ans=0;
    if(mid>=L)ans+=query(L,R,l,mid,rt<<1);
    if(mid<R)ans+=query(L,R,mid+1,r,rt<<1|1);
    return ans;
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n;
    cin>>n;
    build(1,n,1);
    update(9,10,1,n,1,10);
    cout<<query(8,10,1,n,1)<<endl;
}

 

推荐阅读