敌兵布阵 HDU1166

#include<bits/stdc++.h>
using namespace std;

int ans;
struct node
{
       int l,r,w;//l，r分别表示区间左右端点，w表示区间和
}tree[4*50000+1];//字典树要开四倍空间

void build(int l,int r,int k)
{
    tree[k].l=l;tree[k].r=r;
    if(l==r)//叶子节点
    {
        scanf("%d",&tree[k].w);
        return ;
    }
    int m=(l+r)/2;
    build(l,m,k*2);//左孩子
    build(m+1,r,k*2+1);//右孩子
    tree[k].w=tree[k*2].w+tree[k*2+1].w;//状态合并，此结点的w=两个孩子的w之和
}

int ask(int k,int pos)
{
    if(tree[k].l==tree[k].r) //当前结点的左右端点相等，是叶子节点，是最终答案
    {
        return  tree[k].w;
    }
    int m=(tree[k].l+tree[k].r)/2;
    if(pos<=m) ask(k*2,pos);//目标位置比中点靠左，就递归左孩子
    else ask(k*2+1,pos);//反之，递归右孩子
}

void add(int k,int pos,int v)
{
    if(tree[k].l==tree[k].r)//找到目标位置
    {
        tree[k].w+=v;
        return;
    }
    int m=(tree[k].l+tree[k].r)/2;
    if(pos<=m) add(k*2,pos,v);
    else add(k*2+1,pos,v);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;//所有包含结点k的结点状态更新
}


void sum(int k,int left,int right)
{
    if(tree[k].l>=left && tree[k].r<=right)
    {
        ans+=tree[k].w;
        return;
    }
    int m=(tree[k].l+tree[k].r)/2;
    if(left<=m) sum(k*2,left,right);
    if(right>m) sum(k*2+1,left,right);
}

int main()
{

    int cas;cin>>cas;
    int flag=0;
    while(cas--)
    {   printf("Case %d:\n",++flag);
        int n;scanf("%d",&n);
        build(1,n,1);
        char s[2000];
        int x,y;
        while(scanf("%s",s)==1)
        {   ans=0;
            if(s[0]==‘E‘)break;
            scanf("%d%d",&x,&y);
            if(s[0]==‘A‘)add(1,x,y);
            if(s[0]==‘S‘)add(1,x,-y);
            if(s[0]==‘Q‘){sum(1,x,y);printf("%d\n",ans);}
        }
    }
    return 0;
}