[线段树]模板2

【模板】线段树2

 

题目描述

 

给定一个无序数列，有两种操作：

1.将一个区间内的所有点值都加上一个整数

2.求一个区间的和

 

输入

 

输入的第1行，共有两个数n和q，表示数列长度和操作次数

输入的第2行，共有n个数，表示该数列

接下来共有q行，每行有3～4个数 第1个数为操作类型，具体如下

若是第1种操作，接下来三个数x,y,z表示将［x,y]内所有数加上z。

若是第2种操作，接下来两个数x，y表示闭区间的左右端点

 

输出

 

输出共有若干行，对于每一个询问输出一个整数结果

 

样例输入

 

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

 

样例输出

 

10
13
30

 

提示

1<=n,q<=200000

保证输入的所有数据在int范围内

 

 

代码：

#include <cstdio>    
#include <algorithm>    
using namespace std;    
#define lson l , m , rt*2    
#define rson m + 1 , r , rt*2+1 
#define root 1 , N , 1   
const int maxn = 200001;    
long long add[maxn<<2];    
long long sum[maxn<<2];    

void PushUp(int rt) {    
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];    
}
   
void PushDown(int rt, int m) {    
    if (add[rt]) {    
        add[rt << 1] += add[rt];    
        add[rt << 1 | 1] += add[rt];    
        sum[rt << 1] += add[rt] * (m - (m >> 1));    
        sum[rt << 1 | 1] += add[rt] * (m >> 1);    
        add[rt] = 0;    
    }    
}    

void Build(int l, int r, int rt) {    
    add[rt] = 0;    
    if (l == r) {    
        scanf("%lld", &sum[rt]);    
        return ;    
    }    
    int m = (l + r) >> 1;    
    Build(lson);    
    Build(rson);    
    PushUp(rt);    
}    

void Update(int L, int R, int c, int l, int r, int rt) {    
    if (L <= l && r <= R) {    
        add[rt] += c;    
        sum[rt] += (long long)c * (r - l + 1);    
        return ;    
    }    
    PushDown(rt, r - l + 1);    
    int m = (l + r) >> 1;    
    if (L <= m) {
        Update(L, R, c, lson);    
    }
    if (m < R) {
        Update(L, R, c, rson);    
    }
    PushUp(rt);
} 
   
long long Query(int L, int R, int l, int r, int rt) {    
    if (L <= l && r <= R) {    
        return sum[rt];    
    }    
    PushDown(rt, r - l + 1);    
    int m = (l + r) >> 1;    
    long long re = 0;    
    if (L <= m) {
        re += Query(L, R, lson);    
    }
    if (m < R) {
        re += Query(L, R, rson);    
    }
    return re;    
}    

int main() {    
    int N , Q;    
    scanf("%d%d", &N, &Q);    
    Build(root);    
    while (Q --) {    
        char op[2];    
        int a , b , c;    
        if (scanf("%s", op)!= 1) {
            break;  
        }
        if (op[0] == '2') {    
            scanf("%d%d", &a, &b);    
            printf("%lld\n", Query(a , b ,root));    
        } 
        else {    
            scanf("%d%d%d", &a, &b, &c);    
            Update(a, b, c, root);    
        }    
    }    
    return 0;    
}