洛谷p3372 线段树模版

这里写链接内容

#include<cstdio>
using namespace std;
int n,m;
int a[230000];
struct yuy{
long long l,r,len,sum,add;
}s[999999];
void bu(int o,int l,int r)
{
    s[o].l=l;
    s[o].r=r;
    s[o].len=r-l+1;
    if(l==r)
    {
        s[o].sum=a[l];
        return ;
    }

    else{int mid=(l+r)>>1;
    bu(o<<1,l,mid);
    bu((o<<1)|1,mid+1,r);
    s[o].sum=s[o<<1].sum+s[(o<<1)|1].sum;

    }

}
void pushdown(int o)
{
    s[o<<1].add+=s[o].add;
    s[(o<<1)|1].add+=s[o].add;
    s[o<<1].sum+=s[o].add*s[o<<1].len;
    s[(o<<1)|1].sum+=s[o].add*s[(o<<1)|1].len;
    s[o].add=0;
}
void add(int o,int ql,int qr,int ad)
{
    int l=s[o].l,r=s[o].r,m=(l+r)>>1,len=s[o].len;
    if(l==ql&&qr==r)
    {
        s[o].add+=ad;
        s[o].sum+=ad*len;
        return ;
    }
    if(s[o].add) pushdown(o);
    if(qr<=m) add(o<<1,ql,qr,ad);
    else if(ql>m) add((o<<1)|1,ql,qr,ad);
    else add(o<<1,ql,m,ad),add((o<<1)|1,m+1,qr,ad);
    s[o].sum=s[o<<1].sum+s[(o<<1)|1].sum;
}
long long ask(int o,int ql,int qr)
{
    int l=s[o].l,r=s[o].r,m=(l+r)>>1;
    if(l==ql&&qr==r) return s[o].sum;
    if(s[o].add) pushdown(o);
    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()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    bu(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int b,c;
        int t,x;
        scanf("%d",&t);
        if(t==1)
        {
            scanf("%d%d%d",&b,&c,&x);
            add(1,b,c,x);

        }
        if(t==2)
        {
            scanf("%d%d",&b,&c);
            printf("%ld\n",ask(1,b,c));
        }
    }
    return 0;
}