HDU 1166 敌兵布阵（第一个线段树）

题目链接

这个题，用完树状数组，再用线段树水过。线段树，却是感觉比树状数组的功能强多了。只要把线段树的思想理解，然后把实现过程，递归过程了解，单点更新就没问题了。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define N 50001
struct node
{
    int l;
    int r;
    int v;
}tree[4*N];
void build(int l,int r,int rt)
{
    int m;
    tree[rt].l = l;
    tree[rt].r = r;
    if(l == r)
    {
        scanf("%d",&tree[rt].v);
        return;
    }
    m = (l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    tree[rt].v = tree[rt<<1].v + tree[rt<<1|1].v;
}
void update(int pos,int sc,int rt)
{
    int m;
    if(tree[rt].l == tree[rt].r)
    {
        tree[rt].v += sc;
        return ;
    }
    m = (tree[rt].l + tree[rt].r) >> 1;
    if(pos <= m)
    update(pos,sc,rt<<1);
    else
    update(pos,sc,rt<<1|1);
    tree[rt].v = tree[rt<<1].v + tree[rt<<1|1].v;
}
int query(int L,int R,int rt)
{
    int m;
    if(tree[rt].l == L&&tree[rt].r == R)
    {
        return tree[rt].v;
    }
    m = (tree[rt].l + tree[rt].r) >> 1;
    if(L > m)
    {
        return query(L,R,rt<<1|1);
    }
    else if(R <= m)
    {
        return query(L,R,rt<<1);
    }
    else
    {
        return query(L,m,rt<<1)+query(m+1,R,rt<<1|1);
    }
}
int main()
{
    int t,num = 0,x,y,n;
    char str[100];
    scanf("%d",&t);
    while(t--)
    {
        num ++;
        memset(tree,0,sizeof(tree));
        scanf("%d",&n);
        build(1,n,1);
        scanf("%s",str);
        printf("Case %d:\n",num);
        while(strcmp(str,"End") != 0)
        {
            scanf("%d%d%*c",&x,&y);
            if(strcmp(str,"Query") == 0)
            printf("%d\n",query(x,y,1));
            else if(strcmp(str,"Add") == 0)
            update(x,y,1);
            else if(strcmp(str,"Sub") == 0)
            update(x,-y,1);
            scanf("%s",str);
        }
    }
    return 0;
}