P3372 【模板】线段树 1

题目描述

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

将某区间每一个数加上 k。
求出某区间每一个数的和。
输入格式
第一行包含两个整数 n,,m，分别表示该数列数字的个数和操作的总个数。

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

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

1. x y k：将区间 [x, y][x,y] 内每个数加上 k。
2. x y：输出区间 [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;
typedef long long ll;
//#define rh(p) (p << 1) | 1 //定位右孩子
ll n, a[100005], d[270000], b[270000];
void build(ll l, ll r, ll p) {
    if (l == r) {
        d[p] = a[l];
        return;
    }
    ll m = (l + r) >> 1;
    build(l, m, p << 1), build(m + 1, r, (p << 1) | 1);
    d[p] = d[p << 1] + d[(p << 1) | 1];
}
void update(ll l, ll r, ll c, ll s, ll t, ll p) {
    if (l <= s && t <= r) {
        d[p] += (t - s + 1) * c, b[p] += c;
        return;
    }
    ll m = (s + t) >> 1;
    if (b[p])
        d[p << 1] += b[p] * (m - s + 1), d[(p << 1) | 1] += b[p] * (t - m),
        b[p << 1] += b[p], b[(p << 1) | 1] += b[p];
    b[p] = 0;
    if (l <= m) update(l, r, c, s, m, p << 1);
    if (r > m) update(l, r, c, m + 1, t, (p << 1) | 1);
    d[p] = d[p << 1] + d[(p << 1) | 1];
}
ll getsum(ll l, ll r, ll s, ll t, ll p) {
    if (l <= s && t <= r) return d[p];
    ll m = (s + t) >> 1;
    if (b[p])
        d[p << 1] += b[p] * (m - s + 1), d[(p << 1) | 1] += b[p] * (t - m),
        b[p << 1] += b[p], b[(p << 1) | 1] += b[p];
    b[p] = 0;
    ll sum = 0;
    if (l <= m) sum = getsum(l, r, s, m, p << 1);
    if (r > m) sum += getsum(l, r, m + 1, t, (p << 1) | 1);
    return sum;
}
int main() {
    ll q, i1, i2, i3, i4;
    cin >> n >> q;
    for (int i = 1; i <= n; ++i) cin >> a[i];//存放区间数组信息
    build(1, n, 1);
    while (q--) {
        cin >> i1 >> i2 >> i3;
        if (i1 == 2)
            cout << getsum(i2, i3, 1, n, 1) << endl;
        else
            cin >> i4, update(i2, i3, i4, 1, n, 1);  
    }
    return 0;
}