HDU 1166 敌兵布阵 线段树的基本应用——动态区间和问题

题目： http://acm.hdu.edu.cn/showproblem.php?pid=1166

简单题，1A了，这个好像就是传说中的“点树”。

设当前结点表示线段[left, right]，编号为i，则结点的左孩子表示线段[left, mid], 编号为2*i，右孩子表示线段[mid+1, right], 编号为2*i+1。

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

const int MAXN = 50000;

//线段树的结点，分别表示一条线段的左端点、右端点、增加的人数
struct Tree_Node
{
    int left;
    int right;
    int num;
}tree[4*MAXN];

//建立线段树，left和right对应线段树结点中的left和right，i表示当前结点的线段树编号
void build(int left, int right, int i)
{
    tree[i].left = left;
    tree[i].right = right;
    tree[i].num = 0;
    if(left == right)
    {
        return;
    }
    int mid = (left + right) / 2;
    build(left, mid, 2*i);
    build(mid+1, right, 2*i+1);
}

//插入到线段树，tar表示增加人数的那个点在数轴上的编号（不是在线段树上的编号）
//num表示增加的人数，step表示当前结点的线段树编号
void insert(int tar, int num, int step)
{
    if(tree[step].left <= tar && tree[step].right >= tar)
    {
        tree[step].num += num;
    }
    if(tree[step].left == tree[step].right)
    {
        return;
    }
    int mid = (tree[step].left + tree[step].right) / 2;
    if(tar <= mid)
    {
        insert(tar, num, step*2);
    }
    else
    {
        insert(tar, num, step*2+1);
    }
}

//left和right表示查询的区间，step表示当前结点的线段树编号
int query(int left, int right, int step)
{
    if(tree[step].left == left && tree[step].right == right)
    {
        return tree[step].num;
    }
    int mid = (tree[step].left + tree[step].right) / 2;
    if(right <= mid)
    {
        return query(left, right, step*2);
    }
    else if(left > mid)
    {
        return query(left, right, step*2+1);
    }
    else
    {
        return query(left, mid, step*2) + query(mid+1, right, step*2+1);
    }
}

int main()
{
    int t, n, x;
    scanf("%d", &t);
    for(int item = 1; item <= t; item++)
    {
        scanf("%d", &n);
        build(1, n, 1);
        for(int i = 1; i <= n; i++)
        {
            scanf("%d", &x);
            insert(i, x, 1);
        }
        char cmd[20];
        int a, b;
        printf("Case %d:
", item);
        while(scanf("%s", cmd) != EOF && cmd[0] != 'E')
        {
            scanf("%d %d", &a, &b);
            if(cmd[0] == 'Q')
            {
                printf("%d
", query(a, b, 1));
            }
            else if(cmd[0] == 'A')
            {
                insert(a, b, 1);
            }
            else if(cmd[0] == 'S')
            {
                insert(a, -b, 1);
            }
        }
    }
    return 0;
}