线段树模板题

P3372 【模板】线段树 1

#include <iostream>
#include <algorithm>
typedef long long LL;
using namespace std;
const int MAXN = 1000000+5;
LL arr[MAXN];
struct seg { int l, r, lz; LL v; }t[MAXN << 2];//lz延迟标记,l 左边界,r 右边界
int lch(int k) { return k << 1; };
int rch(int k) { return k << 1 | 1; };
inline void push_down(int k) {
    if (!t[k].lz) return;
    int mid = (t[k].l + t[k].r) >> 1, lz = t[k].lz;
    t[lch(k)].lz += lz;
    t[rch(k)].lz += lz;
    t[lch(k)].v += lz * (mid - t[lch(k)].l + 1);
    t[rch(k)].v += lz * (t[rch(k)].r - mid);
    t[k].lz = 0;
}
inline void push_up(int k) { t[k].v = t[lch(k)].v + t[rch(k)].v; }
void build(int k, int l, int r) {//建树
    t[k].l = l; t[k].r = r; t[k].lz = 0;
    if (l == r) {//叶节点
        t[k].v = arr[l];
        return;
    }
    int mid = (l + r) >> 1;
    build(lch(k),l,mid);
    build(rch(k),mid+1,r);
    push_up(k);
}
LL query(int k,int l,int r) {//查询区间[l,r]的和
    if (t[k].l >= l && t[k].r <= r) return t[k].v;
    LL s = 0;
    push_down(k);
    if (t[lch(k)].r >= l) s+=query(lch(k),l,r);
    if (t[rch(k)].l <= r) s+=query(rch(k),l,r);
    return s;
}
void updata(int k, int l,int r,int v) {//区间[l,r]加上v
    if (t[k].l >= l && t[k].r <= r) {
        t[k].v += v * (t[k].r - t[k].l + 1);
        t[k].lz += v;
        return;
    }
    push_down(k);
    if (t[lch(k)].r >= l) updata(lch(k), l, r, v);
    if (t[rch(k)].l <= r) updata(rch(k), l, r, v);
    push_up(k);
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int N, M, a, b, c, d;
    cin >> N >> M;
    for (int i = 1; i <= N; i++) {
        cin >> arr[i];//输入数列
    }
    build(1, 1, N);
    for (int i = 0; i < M; i++) {
        cin >> a;
        if (a == 1) {//修改操作
            cin >> b >> c >> d;
            updata(1, b, c, d);
        }
        else {//查询操作
            cin >> b >> c;
            cout << query(1, b, c) << endl;
        }
    }
    return 0;
}

来源：https://www.cnblogs.com/--zz/p/11456647.html