HDU (线段树 单点更新) 敌兵布阵

哎，又切了一天的水题。

线段树果然必须自己写出来才能叫真正的会了，之前一直在套模板确实不好。

这个题目是单点更新 之 单点增减，=￣ω￣=

#include <cstdio>

const int maxn = (1 << 20);

int n, qL, qR, p, v, sum[maxn];

void build(int o, int L, int R)
{
    if(L == R) { scanf("%d", &sum[o]); return; }
    int M = (L + R) / 2;
    build(o*2, L, M);
    build(o*2+1, M+1, R);
    sum[o] = sum[o*2] + sum[o*2+1];
}

void update(int o, int L, int R)
{
    if(L == R) { sum[o] += v; return; }
    int M = (L + R) / 2;
    if(p <= M) update(o*2, L, M);
    else update(o*2+1, M+1, R);
    sum[o] = sum[o*2] + sum[o*2+1];
}

int query(int o, int L, int R)
{
    if(qL <= L && qR >= R) return sum[o];

    int ans = 0;
    int M = (L +R) / 2;
    if(qL <= M) ans += query(o*2, L, M);
    if(qR > M) ans += query(o*2+1, M+1, R);
    return ans;
}

char cmd[10];

int main()
{
    //freopen("in.txt", "r", stdin);

    int T; scanf("%d", &T);

    for(int kase = 1; kase <= T; kase++)
    {
        printf("Case %d:\n", kase);

        scanf("%d", &n);
        build(1, 1, n);

        while(scanf("%s", cmd) == 1)
        {
            if(cmd[0] == 'E') break;
            if(cmd[0] == 'Q')
            {
                scanf("%d%d", &qL, &qR);
                printf("%d\n", query(1, 1, n));
            }
            else
            {
                scanf("%d%d", &p, &v);
                if(cmd[0] == 'S') v = -v;
                update(1, 1, n);
            }
        }
    }

    return 0;
}