AcWing 243. 一个简单的整数问题2

带懒标记的线段树
注意递归之前要传递懒标记

#include<bits/stdc++.h>
using namespace std;

#define int long long

const int N = 1e5+10;

int n,m;
int a[N];

struct NODE{
    int l,r;
    int sum;
    int lazy;
}tr[N*4];

void pushup(int u){
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}

void pushdown(int u){
    int x=u<<1,y=u<<1|1;
    int lazy=tr[u].lazy;
    tr[x].sum+=lazy*(tr[x].r-tr[x].l+1);
    tr[x].lazy+=lazy;
    tr[y].sum+=lazy*(tr[y].r-tr[y].l+1);
    tr[y].lazy+=lazy;
    tr[u].lazy=0;
}

void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,l,a[l],0};
        return;
    }
    tr[u].l=l,tr[u].r=r;
    int mid=l+r>>1;
    build(u<<1,l,mid);
    build(u<<1|1,mid+1,r);
    pushup(u);
}

int query(int u,int l,int r){
    if(l<=tr[u].l && r>=tr[u].r) return tr[u].sum;
    else{
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(r<=mid) return query(u<<1,l,r);
        else if(l>mid) return query(u<<1|1,l,r);
        else{
            int suml=query(u<<1,l,r),sumr=query(u<<1|1,l,r);
            return suml+sumr;
        }
    }
}

void modify(int u,int l,int r,int d){
    if(l<=tr[u].l && r>=tr[u].r){
        tr[u].sum+=(tr[u].r-tr[u].l+1)*d;
        tr[u].lazy+=d;
    }
    else{
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(r<=mid) modify(u<<1,l,r,d);
        else if(l>mid) modify(u<<1|1,l,r,d);
        else{
            modify(u<<1,l,r,d);
            modify(u<<1|1,l,r,d);
        }
        pushup(u);
    }
}

signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    build(1,1,n);
    char op[2];
    while(m--){
        scanf("%s",op);
        if(*op=='Q'){
            int l,r;
            cin>>l>>r;
            cout<<query(1,l,r)<<endl;
        }
        else{
            int l,r,d;
            cin>>l>>r>>d;
            modify(1,l,r,d);
        }
    }
    return 0;
}