线段树之区间修改模板

#include <iostream>
using namespace std;
const int MAX_N=10010;
int s[4*MAX_N],col[4*MAX_N];
void up(int p){
    s[p]=s[p*2]+s[p*2+1];
}
void down(int p,int l,int r){//延迟标记,与s[p]同时修改
    if(col[p]){
        int mid=(l+r)/2;
        s[p*2]+=col[p]*(mid-l+1);
        s[p*2+1]+=col[p]*(r-mid);
        col[p*2]+=col[p];
        col[p*2+1]+=col[p];
        col[p]=0;
    }
}

void modify(int p,int l,int r,int x,int y,int c){
    if(x<=l&&r<=y){
        s[p]+=(r-l+1)*c;
        col[p]+=c;
        return;
    }
    down(p,l,r);
    int mid=(l+r)/2;
    if(x<=mid){
        modify(p*2,l,mid,x,y,c);
    }
    if(y>mid){
        modify(p*2+1,mid+1,r,x,y,c);
    }
    up(p);
}
int query(int p,int l,int r,int x,int y){
    if(x<=l&&r<=y){
        return s[p];
    }
    down(p,l,r);
    int mid=(l+r)/2,res=0;
    if(x<=mid){
        res+=query(p*2,l,mid,x,y);
    }
    if(y>mid){
        res+=query(p*2+1,mid+1,r,x,y);
    }
    return res;
}
int main() {
    int n;
    cin>>n;
    for(int i=1;i<=n;++i){
        int d;
        cin>>d;
        modify(1,1,n,i,i,d);
    }
    int q;
    cin>>q;
    while(q--){
        int d,x,y,c;
        cin>>d>>x>>y;
        if(d==0){
            cin>>c;
            modify(1,1,n,x,y,c);
        }else{
            cout<<query(1,1,n,x,y)<<endl;
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值