HDU 1166 排兵布阵（线段树单点更新）

题意：

给定n个兵营的士兵初始值， 然后有最多40000个操作；

操作一共有两种， 一个是查询给定[a,b]区间兵营的士兵总和。

另一个是增加/减少指定兵营的士兵数目。

输出每次查询的值。

分析：

线段树单点更新模板

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn =  + ;
struct{
    int l, r, sum;
}tree[maxn*];//起码要*4
int a[maxn];
int n;
void build(int index, int l , int r){
    tree[index].l = l, tree[index].r = r;
    if( l == r){
        tree[index].sum = a[l];//递归建树，当l=r就是叶子节点没办法再分，直接赋值
        return;
    }
    int mid = (l+r) / ;
    build(index * , l, mid);
    build(index *  + , mid +  , r);
    tree[index].sum = tree[index * ].sum + tree[index* + ].sum;//不是叶子节点， 总和等于它儿子的总和
}

void update(int index, int a, int val){ //修改总和的
    if(tree[index].l == a && tree[index].r == a){
        tree[index].sum += val;//找到该点，直接更改
        return;
    }
    int mid = (tree[index].l + tree[index].r) / ;

    if(a <= mid)
        update(index * , a , val);
    else
        update(index *  + , a, val);

    tree[index].sum += val;//路径上的点都要更改。
}

int query(int index , int ql, int qr){

    if(tree[index].l == ql && tree[index].r == qr){
        return tree[index].sum;
    }

    int mid = (tree[index].l + tree[index].r) / ;
    if(qr <= mid) //查询区间只在这个的左子树
        return query(index * , ql, qr);
    if(ql > mid)  //只在右子树
        return query(index *  + , ql, qr);

    return query(index * , ql, mid) + query(index *  + , mid + , qr);//左右子树都有的话， 拆分查询区间
}

int main(){
    int T;
    scanf("%d", &T);
    int kase = ;
    while(T--){
        printf("Case %d:\n", kase++);
        scanf("%d", &n);
        for(int i = ; i <= n; i++){
            scanf("%d", &a[i]);
        }
        build(,,n);
        char cho[];
        while(scanf("%s",cho) && cho[] != 'E'){
            int x, y;
            scanf("%d %d", &x, &y);
            if(cho[] == 'Q'){
                printf("%d\n", query(,x,y));
            }
            else if(cho[] == 'A'){
                update(,x,y);
            }
            else update(,x,-y);
        }
    }
    return ;
}