敌兵布阵

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

#include <bits/stdc++.h>
#define lid id<<1
#define rid id<<1|1
#define maxn 50005
using namespace std;
struct node
{
    int l,r,num;
}tr[maxn*4];
int s[maxn];

void push_up(int id)
{
    tr[id].num=tr[lid].num+tr[rid].num;
}
void build(int id,int l,int r)
{
    tr[id].l=l;tr[id].r=r;
    if(l==r)
    {
        tr[id].num=s[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(lid,l,mid);
    build(rid,mid+1,r);
    push_up(id);
}
void update(int id,int l,int r,int add)
{
    if(tr[id].l==l&&tr[id].r==r)
    {
        tr[id].num+=add;
        return ;
    }
    int mid=(tr[id].l+tr[id].r)>>1;
    if(l<=mid) update(lid,l,r,add);
    else update(rid,l,r,add);
    push_up(id);
}
int  query(int id,int l,int r)
{
    if(tr[id].l>=l&&tr[id].r<=r)
    {
        return tr[id].num;
    }
    int mid=(tr[id].l+tr[id].r)>>1;
    if(r<=mid) return query(lid,l,r);
    else if(l>mid) return query(rid,l,r);
    else return query(lid,l,mid)+query(rid,mid+1,r);
}
int main()
{
    int t;
    scanf("%d",&t);
    int flag=0;
    while(t--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",&s[i]);
        build(1,1,n);
        char s[10];
        printf("Case %d:
",++flag);
        while(~scanf("%s",s))
        {
            int l,r,add;
            if(s[0]=='E') break;
            else if(s[0]=='A')
            {
                scanf("%d%d",&l,&add);
                update(1,l,l,add);
            }
            else if(s[0]=='S')
            {
                scanf("%d%d",&l,&add);
                update(1,l,l,-add);
            }
            else
            {
                scanf("%d%d",&l,&r);
                int ans=query(1,l,r);
                printf("%d
",ans);
            }
        }
    }
    return 0;
}