HDU1166 - 敌兵布阵

题目大意

给定N个数，可以对这些数进行三种操作：

1、Add(i,x)，对第i个数增加x

2、Sub(i,x)，对第i个数减x

3、Query(x,y) ，查询区间[x,y]的和

题解

赤裸裸的树状数组啊。。。我还二逼得WA了两次，原因是读入字符串的时候用的while(cin>>s)，然后就一直不能退出循环了，不知道为啥会这样。。。换成C语言风格的字符串，然后用scanf读入就没问题了。。尼玛坑爹啊。。。

代码：

#include<iostream>

#include<cstdio>

#include<cstring>

#define MAXN 50005

using namespace std;

int c[MAXN];

int  n;

int lowbit(int x)

{

    return x&-x;

}

void add(int  x,int d)

{

    while(x<=n)

    {

        c[x]+=d;

        x+=lowbit(x);

    }

}

int  sum(int x)

{

    int  ret=0;

    while(x>0)

    {

        ret+=c[x];

        x-=lowbit(x);

    }

    return ret;

}

int main(void)

{

    int i,j,T,x,y,p=0;

    char s[10];

    cin>>T;

    while(T--)

    {

        printf("Case %d:\n",++p);

        scanf("%d",&n);

        memset(c,0,sizeof(c));

        for(i=1; i<=n; i++)

        {

            scanf("%d",&j);

            add(i,j);

        }

        while(scanf("%s",s)!=EOF)

        {

            if(s[0]=='E') break;

            scanf("%d%d",&x,&y);

            if(s[0]=='Q')

                printf("%d\n",sum(y)-sum(x-1));

            else if (s[0]=='A')

                add(x,y);

            else

                add(x,-y);

        }

    }

    return 0;

}