【模板】线段树1

17.【模板】线段树1

题目描述

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

1. 将某区间每一个数加上 k。
2. 求出某区间每一个数的和。

输入格式

第一行包含两个整数 n,m，分别表示该数列数字的个数和操作的总个数。

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

接下来 mm 行每行包含 33 或 44 个整数，表示一个操作，具体如下：

1. 1 x y k：将区间 [x,y] 内每个数加上 k。
2. 2 x y：输出区间 [x,y] 内每个数的和。

输出格式

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

输入输出样例

输入 #1复制

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

输出 #1复制

11
8
20

说明/提示

【样例解释】

题解

#include<iostream>
using namespace std;
#define maxn 500005
#define ll long long
typedef struct
{
    int l,r;
    ll sums,lz;
}node;
node a[maxn];
ll c[maxn];
int n,m;
void build(int i,int l,int r)
{
    a[i].l=l,a[i].r=r;
    if(l==r) {
        a[i].sums=c[l];
        return ;}
    int mid=(l+r)>>1;
    build(2*i,l,mid);
    build(2*i+1,mid+1,r);
    a[i].sums=a[i*2].sums+a[i*2+1].sums; 
}

void pushdown(int x)
{
    if(a[x].lz!=0)
    {
        a[2*x].lz+=a[x].lz;
        a[2*x+1].lz+=a[x].lz;
        int mid=(a[x].r+a[x].l)>>1;
        a[2*x].sums+=a[x].lz*(mid-a[2*x].l+1);
        a[2*x+1].sums+=a[x].lz*(a[2*x+1].r-mid);
        a[x].lz=0;
    }
    return ;
}
void update(int i,ll k,int l,int r)
{
    if(l<=a[i].l&&a[i].r<=r)
    {
        a[i].sums+=k*(a[i].r-a[i].l+1);
        a[i].lz+=k;
        return ;
    }
    pushdown(i);
    if(a[2*i].r>=l) update(2*i,k,l,r);
    if(a[2*i+1].l<=r) update(2*i+1,k,l,r);
    a[i].sums=a[2*i].sums+a[2*i+1].sums;
    return ;
}

ll getsum(int i,int l,int r)
{
    if(a[i].l>=l&&a[i].r<=r)
    {
        return a[i].sums;
    }
    ll res=0;
    pushdown(i);
    if(a[2*i].r>=l) res+=getsum(2*i,l,r);
    if(a[2*i+1].l<=r) res+=getsum(2*i+1,l,r);
    return res;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>c[i];
    }
    build(1,1,n);
    int p1,x,y; ll k;
    for(int i=1;i<=m;i++)
    {
        cin>>p1;
        switch(p1)
        {
            case 1:cin>>x>>y>>k;update(1,k,x,y);break;
            case 2:cin>>x>>y;cout<<getsum(1,x,y)<<endl;break;
            default:break;
        }
    }
    return 0;
}