线段树区间更新模板 POJ - 3468

题目链接：http://poj.org/problem?id=3468

两种操作：

1. 对区间[a ,b]中的每一个数字增加c
2. 查询区间[a, b]中的数字和

顺便提一下lazy的思想，因为对于区间更新来说，一个区间中的每一个点增加或者减少某一个值，如果每一次都实现完全更新会浪费许多时间，经过观察不难发现，假如有一个[1, 7]的区间， 第一次对[4, 7]增加10，第二次对[4, 5]减少5，我们可以在第一次更新[4, 7]时对这个区间打上一个Lazy标记，其值为10， 直接对区间每一个值加10，而不继续向下深入到每一个节点。 第二次更新[4, 5]时为了找到[4, 5]这个区间，必定会经过[4, 7]这个区间，只需要遍历时检查经过区间内的标记是否为0， 如果不为0则将lazy标记向下深入实现，在这个例子中，我们遍历到[4, 7]时发现第一次更新时留下的值为10的标记，于是向下推， 将[4, 5]， [6, 7]区间lazy标记为10， 直接对区间增加区间长度个10，然后再对于[4, 5]以同样的方法实现第二次减5这个更新。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

#define LL long long
#define lson ins<<1
#define rson ins<<1|1
#define merge tree[ins]=tree[lson]+tree[rson]
#define mid (r+l)/2

using namespace std;

const int Max = 100052;

struct Tree {
    int l, r;
    LL sum, lazy;

    void updata(LL x) {//更新某一个节点
        sum += (r - l + 1) * x;
        lazy += x;
    }
} tree[Max << 2];

LL data[Max << 2];

inline void push_down(int ins) {//lazy标记向下传递
    LL lazyval = tree[ins].lazy;
    if (lazyval) {
        tree[lson].updata(lazyval);
        tree[rson].updata(lazyval);
        tree[ins].lazy = 0;
    }
}

inline void push_up(int ins) {//可以通过改变这个函数实现许多不同的功能
    tree[ins].sum = tree[lson].sum + tree[rson].sum;
}

inline void build(int l, int r, int ins) {
    tree[ins].l = l, tree[ins].r = r, tree[ins].sum = tree[ins].lazy = 0;
    if (r == l) {
        tree[ins].sum = data[r];
    } else if (r > l) {
        build(l, mid, lson);
        build(mid + 1, r, rson);
        push_up(ins);
    }
}

inline void updata_tree(int ql, int qr, LL x, int ins) {//更新
    int l = tree[ins].l, r = tree[ins].r;
    if (ql <= l && r <= qr)tree[ins].updata(x);
    else {
        push_down(ins);
        if (ql <= mid)updata_tree(ql, qr, x, lson);
        if (qr > mid)updata_tree(ql, qr, x, rson);
        push_up(ins);
    }
}

inline LL query(int ql, int qr, int ins) {//查询
    int l = tree[ins].l, r = tree[ins].r;
    if (ql <= l && r <= qr)return tree[ins].sum;
    else {
        push_down(ins);
        LL res = 0;
        if (ql <= mid)res += query(ql, qr, lson);
        if (qr > mid)res += query(ql, qr, rson);
        return res;
    }
}

int main() {
    int n, m;
    while (~scanf("%d%d", &n, &m)) {
        for (int a = 1; a <= n; a++) {
            scanf("%lld", &data[a]);
        }
        build(1, n, 1);
        char opera[3];
        for (int a = 1; a <= m; a++) {
            scanf("%s", opera);
            if (opera[0] == 'Q') {
                int i, j;
                scanf("%d%d", &i, &j);
                printf("%lld\n", query(i, j, 1));
            } else {
                int i, j;
                LL x;
                scanf("%d%d%lld", &i, &j, &x);
                updata_tree(i, j, x, 1);
            }
        }
    }
    return 0;
}