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|1int 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;}