hdu 敌兵布阵 线段树基础

/*第一道线段树...参考了not only success的代码风格,挺好的.

这是一道单点更新的线段树.先修改叶子,然后递归修改祖先.也可用树状数组.*/

#include <stdio.h>
#include <cstring>
#define maxn 50001
#define lson l,mid,rt<<1//<<箭头指向的为移动的方向
#define rson mid+1,r,rt<<1|1
int sum[maxn<<2];
int n,l,r;
void pushup(int rt)//将信息传递到父亲节点
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];//左移一维以后和1取并
}
void build(int l,int r,int rt)//rt为祖先的数组下标
{
    if(l==r) //叶子节点的值读入,叶子节点线段单元的权值,该区域内的人数,重量之类.
    {
        scanf("%d",&sum[rt]);
        return ;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int p,int val,int l,int r,int rt)//p为要修改的线段元的位置,val为要改变的值,后面意义同上.类比树状数组理解.
{
    //先对叶子节点进行修改,然后递归修改它的祖先
    if(l==r)
    {
        sum[rt]+=val;
        return ;
    }
    int mid=(r+l)>>1;//p<mid,就更新左子树,否则,更新右子树
    if(p<=mid) update(p,val,lson);
    else update(p,val,rson);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r) return sum[rt];//这是所要查询的线段完全包含当前线段的情况,满足题意
    int mid=(l+r)>>1;
    int s=0;//这是不完全包含的情况
    if(R>mid) s+=query(L,R,rson);//将查询的区间拆成子区间进行查询,最后返回的子区间的和即为祖先的值,待查询的值.
    if(L<=mid) s+=query(L,R,lson);//第二个if的前面不能加else.因为这两个if可能是同时满足的.
    return s;
}
int main()
{
    int t,cases=1,a,b;
    scanf("%d",&t);
    while(t--)
    {
        printf("Case %d:\n",cases++);
        scanf("%d",&n);
        build(1,n,1);
        char op[10];
        memset(op,0,sizeof(op));
        while(scanf("%s",op)==1&&strcmp(op,"End")!=0)
        {
            scanf("%d%d",&a,&b);
            if(op[0]=='A')
                update(a,b,1,n,1);
            else if(op[0]=='S')
                update(a,-b,1,n,1);
            else
                printf("%d\n",query(a,b,1,n,1));
        }
    }
    return 0;
}