codevs p1082线段树 区间修改

lazy表示的是给节点o的lc和rc加减的数字

也是codevs p1082的题解

#include <bits/stdc++.h>
#define lc (o<<1)
#define rc (lc|1)
#define m ((l+r)>>1)
using namespace std;
const int MAXN = 200000 + 5;
typedef long long LL;
LL lazy[MAXN<<2],sum[MAXN<<2],a[MAXN];
int in()
{
    LL c=getchar(),x=0;
    while(!isdigit(c))c=getchar();
    while(isdigit(c)){x=x*10+c-'0';c=getchar();}
    return x;
}
inline void up(int o,int l,int r)
{
    sum[o]=0;
    if(l==r)sum[o]=a[r];
    else sum[o]=sum[lc]+sum[rc];
    if(lazy[o])sum[o]+=lazy[o]*(r-l+1);
}
void update(int o,int l,int r,int cl,int cr,int cv)
{
    if(cl<=l && r<=cr)
    {
        lazy[o]+=cv;
    }
    else
    {
        if(cl<=m)update(lc,l,m,cl,cr,cv);
        if(m+1<=cr)update(rc,m+1,r,cl,cr,cv);
    }
    up(o,l,r);
}
LL query(int o,int l,int r,int ql,int qr,LL add)
{
    LL ans=0;
    if(ql<=l && r<=qr)
    {
        return sum[o]+add*(r-l+1);
    }
    if(ql<=m)ans+=query(lc,l,m,ql,qr,add+lazy[o]);
    if(m+1<=qr)ans+=query(rc,m+1,r,ql,qr,add+lazy[o]);
    return ans;
}
void build(int o,int l,int r)
{
    if(l==r)
    {
        sum[o]=a[r];
        return;
    }
    build(lc,l,m);
    build(rc,m+1,r);
    up(o,l,r);
}
int main()
{
    int n=in();
    for(int i=1;i<=n;i++)a[i]=in();
    build(1,1,n);
    /*
    printf("\nbug\n\n");
    for(int i=1;i<=n;i++)printf("%d ",query(1,1,n,i,i,0));
    printf("\nbug\n\n");
    */
    int q=in();
    while(q--)
    {
        int type=in();
        if(type==1)
        {
            int x=in(),y=in(),z=in();
            update(1,1,n,x,y,z);
        }
        else
        {
            int x=in(),y=in();
            printf("%lld\n",query(1,1,n,x,y,0));
        }
        /*
        printf("\nbug\n\n");
        for(int i=1;i<=2*n-1;i++)printf("%lld ",lazy[i]);
        printf("\n");
        for(int i=1;i<=n;i++)printf("%lld ",query(1,1,n,i,i,0));
        printf("\nbug\n\n");
        */
    }
    return 0;
}