hdu1166 敌兵布阵（线段树点更新模板题）

题目链接：点击打开链接

大意：敌人有N个营地，一开始知道每个营地的人数。

            接下来会询问四种命令：(1) Add i j，i 和 j 为正整数，表示第i个营地增加j个人（j不超过30）；
                                                        (2) Sub i j ，i 和 j 为正整数，表示第i个营地减少j个人（j不超过30）；
                                                        (3) Query i j , i 和 j 为正整数，i <= j，表示询问第i到第j个营地的总人数；
                                                        (4) End 表示结束，这条命令在每组数据最后出现。

            遇到Query则作出回答，遇到End则结束本次数据的询问。其他则修改相关营地的数据。（显然，Sub 可以归并到 Add 操作）

思路：

对于每一组数据，用build初始化线段树，每个结点保存区间[i, j]内的总人数。叶子结点即是每个营地的人数，这个由输入给出，对于内结点，则采取pushUp由下往上传递值。

对于Add 操作（包括Sub操作），可以用update来修改叶子结点的值，与build相似，内结点采取pushUp更新其值。

对于Query 操作，用query来查看区间的人数。对于没有在一个存储区间表示出来的空间，则采取同时向两边走，覆盖后返回其值的方式取值，详细见代码。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int maxn = 50000 + 10;
int sum[maxn<<2];
void pushUp(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l, int r, int rt)
{
    if(l == r)
    {
        scanf("%d", &sum[rt]);
        return;
    }
    int m = (l+r)>>1;
    build(lson);
    build(rson);
    pushUp(rt);
}
void update(int p, int add, int l, int r, int rt)
{
    if(l == r)
    {
        sum[rt] += add;
        return;
    }
    int m = (l+r)>>1;
    if(p <= m)
        update(p, add, lson);
    else
        update(p, add, rson);
    pushUp(rt);
}
int query(int L, int R, int l, int r, int rt)
{
    if(L<=l && r<=R)
        return sum[rt];
    int m = (l+r)>>1;
    int ans = 0;
    if(L <= m)
        ans += query(L, R, lson);
    if(R > m)
        ans += query(L, R, rson);
    return ans;
}
int main()
{
    int T, n, a, b;
    char sa[10];
    scanf("%d", &T);
    for(int i = 1; i <= T; i++)
    {
        printf("Case %d:\n", i);
        scanf("%d", &n);
        build(1, n, 1);
        while(scanf("%s", sa), sa[0] != 'E')
        {
            scanf("%d%d", &a, &b);
            if(sa[0] == 'Q')
                printf("%d\n",query(a, b, 1, n, 1));
            else if(sa[0] == 'A')
                update(a, b, 1, n, 1);
            else
                update(a, -b, 1, n, 1);
        }
    }
    return 0;
}