[LG]p3372线段树模板（区间修改+求和）

题目链接

题目描述：

如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数加上x

2.求出某区间每一个数的和

输入格式：

第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和

输出格式：

输出包含若干行整数，即为所有操作2的结果。

线段树模板，没什么好说的

建树:

struct Node
{
    int l,r,len;
    ll val;
    int lazy;//sign
}tree[N*2];
void Maketree(int o,int l,int r)
{
    tree[o].l=l,tree[o].r=r,tree[o].len=r-l+1;
    if(l==r) tree[o].val=data[l];
    else
    {
        int mid=(l+r)>>1,nxt=o<<1;
        Maketree(nxt,l,mid);
        Maketree(nxt+1,mid+1,r);
        tree[o].val=tree[nxt].val+tree[nxt+1].val;
    }
}

更新数据:

void pushdown(int o)//将lazy标记下传到左右子树
{
    if(!tree[o].lazy) return ;
    int p=tree[o].lazy,nxt=o<<1;
    tree[nxt].lazy+=p;
    tree[nxt+1].lazy+=p;
    tree[nxt].val+=p*tree[nxt].len;
    tree[nxt+1].val+=p*tree[nxt+1].len;
    tree[o].lazy=0;
}

在[l,r]加上addval:

void update(int o,int l,int r,int addval)
{
    if(l<=tree[o].l && r>=tree[o].r)
    {
        tree[o].lazy+=addval;
        tree[o].val+=tree[o].len*addval;
        return ;
    }
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(l<=mid) update(nxt,l,min(r,mid),addval);
    if(r>mid) update(nxt+1,max(l,mid+1),r,addval);
    tree[o].val=tree[nxt].val+tree[nxt+1].val;
}

查询：

ll query(int o,int l,int r)
{
    if(tree[o].l==l && tree[o].r==r) return tree[o].val;
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(r<=mid) return query(nxt,l,r);
    else if(l>mid) return query(nxt+1,l,r);
    return query(nxt,l,mid)+query(nxt+1,mid+1,r);
}

完整程序（丑到不行）：

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int n,m;
const int SIZE=1<<15;
inline char getch(){
    static char buf[SIZE],*p1=buf,*p2=buf;
    return p1==p2 && (p2=(p1=buf)+fread(buf,1,SIZE,stdin),p1==p2) ? EOF : *p1++;
}
int read(){
    int s=0,f=1;
    register char c=getch();
    while(c>'9'|| c<'0') { if(c=='-') f=-1;c=getch();}
    while(c<='9' && c>='0') {s=(s<<3)+(s<<1)+c-'0';c=getch();}
    return s*=f;
}

struct Node
{
    int l,r,len;
    ll val;
    int lazy;//sign
}tree[N*2];
int data[N];
void Maketree(int o,int l,int r)
{
    tree[o].l=l,tree[o].r=r,tree[o].len=r-l+1;
    if(l==r) tree[o].val=data[l];
    else
    {
        int mid=(l+r)>>1,nxt=o<<1;
        Maketree(nxt,l,mid);
        Maketree(nxt+1,mid+1,r);
        tree[o].val=tree[nxt].val+tree[nxt+1].val;
    }
}

void pushdown(int o)
{
    if(!tree[o].lazy) return ;
    int p=tree[o].lazy,nxt=o<<1;
    tree[nxt].lazy+=p;
    tree[nxt+1].lazy+=p;
    tree[nxt].val+=p*tree[nxt].len;
    tree[nxt+1].val+=p*tree[nxt+1].len;
    tree[o].lazy=0;
}

void update(int o,int l,int r,int addval)
{
    if(l<=tree[o].l && r>=tree[o].r)
    {
        tree[o].lazy+=addval;
        tree[o].val+=tree[o].len*addval;
        return ;
    }
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(l<=mid) update(nxt,l,min(r,mid),addval);
    if(r>mid) update(nxt+1,max(l,mid+1),r,addval);
    tree[o].val=tree[nxt].val+tree[nxt+1].val;
}
ll query(int o,int l,int r)
{
    if(tree[o].l==l && tree[o].r==r) return tree[o].val;
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(r<=mid) return query(nxt,l,r);
    else if(l>mid) return query(nxt+1,l,r);
    return query(nxt,l,mid)+query(nxt+1,mid+1,r);
}

int main()
{
    int opt,x,y,k;
    n=read(),m=read();
    for(int i=1;i<=n;i++)
        data[i]=read();
    Maketree(1,1,n);
    while(m--)
    {
        opt=read();
        if(opt==1)
        {
            x=read(),y=read(),k=read();
            update(1,x,y,k);
        }
        if(opt==2)
        {
            x=read(),y=read();
            printf("%lld\n",query(1,x,y));
        }
//		for(int i=1;tree[i].val!=0;i++)
//		    printf("tree[%d].val=%d ",i,tree[i].val);
//		printf("\n");
    }
    return 0;
}