线段树区间修改与查询（求和）

P3372

#include<iostream> 
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#define MAXN 200000
#define LL long long
using namespace std;
struct H{
    LL addi,sum,l,r,len;
}st[MAXN*4+5];
LL a0[MAXN+5];
void build(int o,int l,int r)
{
    st[o].l=l,st[o].r=r,st[o].len=r-l+1;
    if(l==r)
    {
        st[o].sum=a0[l];
        return; 
    }
    else
    {
        int m=(l+r)>>1;
        build(o<<1,l,m);
        build((o<<1)|1,m+1,r);
        st[o].sum=st[o<<1].sum+st[(o<<1)|1].sum;
    }
}
void push(int o)
{
    st[o<<1].addi+=st[o].addi;
    st[(o<<1)|1].addi+=st[o].addi;
    st[o<<1].sum+=st[o].addi*st[o<<1].len;
    st[(o<<1)|1].sum+=st[o].addi*st[(o<<1)|1].len;//push(o<<1),push((o<<1)|1);不用！下一层即可
    st[o].addi=0;
}
void add(int o,int a,int b,int ad)
{
    int l=st[o].l,r=st[o].r,len=st[o].len;
    int m=(l+r)>>1;
    if(l==a&&b==r)
      {st[o].sum+=len*ad,st[o].addi+=ad;
       return;
      }

    if(st[o].addi) push(o);

    if(b<=m)
      add(o<<1,a,b,ad);
    else if(a>m)
      add((o<<1)|1,a,b,ad);
    else//if(a<m&&b>m)×
      add(o<<1,a,m,ad),add((o<<1)|1,m+1,b,ad);
    st[o].sum=st[o<<1].sum+st[(o<<1)|1].sum;
}
LL ask(int o,int ql,int qr)
{
    int l=st[o].l,r=st[o].r;
    if(ql>r||qr<l)return 0;
    if(ql<=l&&qr>=r)return st[o].sum;
    if(st[o].addi) push(o);
    int m=(l+r)>>1;
    if(qr<=m)return ask((o<<1),ql,qr);
    else if(ql>m)return ask((o<<1)|1,ql,qr);

    else
    return ask((o<<1),ql,m)+ask((o<<1)|1,m+1,qr);


}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
     scanf("%lld",&a0[i]);

    build(1,1,n);

    for(int i=1;i<=m;i++)
    {
        int p,x,y,k;
        scanf("%d%d%d",&p,&x,&y);
        if(p==1)
        {
            scanf("%d",&k);
            add(1,x,y,k);
        }
        else
        {
            printf("%lld
",ask(1,x,y));
        }
    }
    return 0;
}

前辈模板

#include<cstdio>
using namespace std;
struct tree{
    long long addi,sum,l,r,len;
}tr[800000];
int n,q,d[200001];

void build(int x,int l,int r){
    tr[x].l=l,tr[x].r=r,tr[x].len=r-l+1;
    if(l==r){tr[x].sum=d[l];return ;}
    int m=(l+r)>>1;
    build(x<<1,l,m);
    build(x<<1|1,m+1,r);
    tr[x].sum=tr[x<<1].sum+tr[x<<1|1].sum;
}

void pushdown(int x){
    tr[x<<1].addi+=tr[x].addi;
    tr[x<<1|1].addi+=tr[x].addi;
    tr[x<<1].sum+=tr[x].addi*tr[x<<1].len;
    tr[x<<1|1].sum+=tr[x].addi*tr[x<<1|1].len;
    tr[x].addi=0;
}

void add(int a,int b,int ad,int x){
    int l=tr[x].l,r=tr[x].r,m=(tr[x].l+tr[x].r)>>1,len=tr[x].len;
    if(l==a&&r==b)    {
        tr[x].addi+=ad;
        tr[x].sum+=len*ad;
        return ;
    }
    if(tr[x].addi)    pushdown(x);
    if(b<=m)    add(a,b,ad,x<<1);
    else if(a>m)    add(a,b,ad,x<<1|1);
    else add(a,m,ad,x<<1),add(m+1,b,ad,x<<1|1);
    tr[x].sum=tr[x<<1].sum+tr[x<<1|1].sum;
}


long long ask(int a,int b,int x){
    int l=tr[x].l,r=tr[x].r,m=(tr[x].l+tr[x].r)>>1;
    if(a==l&&b==r){return tr[x].sum;}
    if(tr[x].addi)    pushdown(x);
    if(b<=m)    return ask(a,b,x<<1);
    else if(a>m)    return ask(a,b,x<<1|1);
    else return ask(a,m,x<<1)+ask(m+1,b,x<<1|1);
}

int main(){
    int i,j,a,b,x;
    scanf("%d",&n);
    for(i=1;i<=n;++i)
        scanf("%d",&d[i]);
    build(1,1,n);
    scanf("%d",&q);
    for(i=1;i<=q;++i){
        scanf("%d",&j);
        if(j==1){
            scanf("%d%d%d",&a,&b,&x);
            add(a,b,x,1);
        }
        else     {
            scanf("%d%d",&a,&b);
            printf("%ld
",ask(a,b,1));
        }
    }
    return 0;
}