洛谷-3372 【模板】线段树 1

题目描述
如题，已知一个数列，你需要进行下面两种操作：
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的结果。

输入输出样例
输入样例#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

说明
时空限制：1000ms,128M
数据规模：
对于30%的数据：N<=8，M<=10
对于70%的数据：N<=1000，M<=10000
对于100%的数据：N<=100000，M<=100000
（数据已经过加强^_，保证在int64/long long数据范围内）

#include
#include
#define N 800044
using namespace std;
long long tree[N]={0};
long long lazy[N]={0};
void push(int root,int l,int r){
    if(!lazy[root]) return;
    int mid=(l+r)/2;
    lazy[2*root]+=lazy[root];
    lazy[2*root+1]+=lazy[root];
    tree[2*root]+=(mid-l+1)*lazy[root];
    tree[2*root+1]+=(r-mid)*lazy[root];
    lazy[root]=0;
}
void update(int root,int l,int r,int L,int R,int val){
    push(root,l,r);
    if(L<=l&&r<=R){
        tree[root]+=(r-l+1)*val;
        lazy[root]+=val;
        return;
    }
    int mid=(l+r)/2;
    if(L<=mid) update(2*root,l,mid,L,R,val);
    if(R>mid) update(2*root+1,mid+1,r,L,R,val);
    push(root,l,r);
    tree[root]=tree[2*root]+tree[2*root+1];
}
long long query(int root,int l,int r,int L,int R){
    push(root,l,r);
    if(L<=l&&r<=R) return tree[root];
    int mid=(l+r)/2;
    push(root,l,r);
    long long sum=0;
    if(L<=mid) sum+=query(2*root,l,mid,L,R);
    if(R>mid)  sum+=query(2*root+1,mid+1,r,L,R);
    return sum;
}
int n=0,m=0;
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        int a=0;cin>>a;
        update(1,1,n,i,i,a);
    }
    while(m--){
        int cmd;cin>>cmd;
        if(cmd==1){
            int a=0,b=0,c=0;cin>>a>>b>>c;
            update(1,1,n,a,b,c);
        }else{
            int a,b;cin>>a>>b;
            cout<