POJ 3468 A Simple Problem with Integers 线段树

题目链接：

poj: http://poj.org/problem?id=3468

题意：

成段更新，成段查询。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LSON o<<1
#define RSON (o<<1)+1
#define M l+(r-l)/2
using namespace std;

typedef long long LL;
const int maxn = (100000 << 2) + 10;

int _n, _q;
LL _sumv[maxn],_addv[maxn];
//bool tag[maxn];

void pushdown(int o, int l, int r){
    if (r > l){
        _addv[LSON] += _addv[o];
        _addv[RSON] += _addv[o];
        _addv[o] = 0;
    }
}

void maintain(int o, int l, int r){
    _sumv[o] = 0;
    if (r > l){
        _sumv[o] = _sumv[LSON] + _sumv[RSON];
    }
    _sumv[o] += _addv[o] * (r - l + 1);
}

void build(int o, int l, int r){
    if (l == r){
        scanf("%lld", _addv + o);
    }
    else{
        build(LSON, l, M);
        build(RSON, M + 1, r);
    }
    maintain(o, l, r);
}

int _ul, _ur, _v;
void update(int o, int l, int r){
    if (_ul <= l&&r <= _ur){
        _addv[o] += _v;
    }
    else{
        pushdown(o, l, r);
        if (_ul <= M) update(LSON, l, M); else maintain(LSON, l, M);
        if (_ur > M) update(RSON, M + 1, r); else maintain(RSON, M + 1, r);
    }
    maintain(o, l, r);
}

int _ql, _qr;
LL _sum;
void query(int o, int l, int r){
    if (_ql <= l&&r <= _qr){
        _sum += _sumv[o];
    }
    else{
        pushdown(o, l, r);
        maintain(LSON, l, M);
        maintain(RSON, M + 1, r);
        if (_ql <= M) query(LSON, l, M);
        if (_qr > M) query(RSON, M + 1, r);
    }
}


void init(){
    memset(_addv, 0, sizeof(_addv));
}

int main(){
    while (scanf("%d%d", &_n, &_q) == 2 && _n){
        init();
        build(1, 1, _n);
        char cmd[2];
        while (_q--){
            scanf("%s", cmd);
            if (cmd[0] == 'Q'){
                _sum = 0;
                scanf("%d%d", &_ql, &_qr);
                query(1, 1, _n);
                printf("%lld
", _sum);
            }
            else{
                scanf("%d%d%d", &_ul, &_ur, &_v);
                update(1, 1, _n);
            }
        }
    }
    return 0;
}