HDOJ 1166 敌兵布阵

线段树的入门题，数组代替树；

线段树是一种完全二叉树，主要操作有建树、修改节点、区间询问（最大值或者和等），由于采用二叉树结构，对一个结点的操作复杂为 logn；

这里建树、修改和询问操作没有采用递归方式，主要是利用了线段树也是完全二叉树的特点，因此可以直接根据要查看节点的编号（1..n）求得其对应在树中的编号，就可以直接进行操作，而不需要递归来实现。

# include <stdio.h>
# include <string.h>

# define N (1 << 17)
# define M ((1 << 16) )

int segSum[N];

void initTree(void)
{
    segSum[0] = 0;

    memset(segSum, 0, sizeof(segSum));
}

int query(int Left, int Right)
{
    int ret;

    ret = 0;
    Left += M-1, Right += M+1;
    while ((Right ^ Left) != 1)
    {        
        if ((Left&0x1) == 0) ret += segSum[Left+1];
        if ((Right&0x1) == 1) ret += segSum[Right-1];
        Left >>= 1, Right >>= 1;
    }

    return ret;
}

void add(int i, int dx)
{
    int p;

    p = i + M;
    while (p != 0)
    {
        segSum[p] += dx;
        p >>= 1;
    }
}

int main()
{
    char buf[10];
    int T, n, i, j, x, k;

    scanf("%d", &T);
    for (k = 1; k <= T; ++k)
    {
        initTree();
        scanf("%d", &n);
        for (i = 1; i <= n; ++i)
        {
            scanf("%d", &x);
            add(i, x);
        }
        printf("Case %d:\n", k);
        while (1)
        {
            scanf("%s", buf);
            if (buf[0] == 'E') break;
            scanf("%d%d", &i, &j);
            if (buf[0] == 'Q') printf("%d\n", query(i, j));
            else if (buf[0] == 'A') add(i, j);
            else if (buf[0] == 'S') add(i, -j);
        }
    }

    return 0;
}

//