HDU 1166 敌兵布阵

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

线段树，单点增减，区间求和

#include <stdio.h>
#include <string.h> 
#include <stdlib.h>
#include <math.h>
#define lson l,m,rt<<1 
#define rson m+1,r,rt<<1|1 
const int maxn=51000; 
int sum[maxn<<2]; 
int nCase=1; 
int pushup(int rt) 
{
    return sum[rt]=sum[rt<<1]+sum[rt<<1|1]; 
}
void build(int l,int r,int rt) 
{
    if(l==r){ 
        scanf("%d",&sum[rt]); 
        return ; 
    }
    int m=(l+r)>>1; 
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int p,int add,int l,int r,int rt) 
{
    if(l==r) 
    {
        sum[rt]+=add;
        return; 
    }
    int m=(l+r)>>1; 
    if(p<=m)update(p,add,lson); 
    else update(p,add,rson); 
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt) 
{
    if(L<=l&&r<=R){ 
        return sum[rt]; 
    }
    int m=(l+r)>>1; 
    int ans=0; 
    if(L<=m)ans+=query(L,R,lson); 
    if(R>m)ans+=query(L,R,rson); 
    return ans; 
}
int main() 
{
    int t,n; 
    scanf("%d",&t); 
    while(t--) 
    {
        scanf("%d",&n); 
        build(1,n,1); 
        char op[10]; 
        printf("Case %d:\n",nCase++); 
        while(scanf("%s",op)) 
        {
            if(op[0]=='E')break; 
            int a,b; 
            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; 
}