HDU 1166 敌兵布阵 树状数组

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

 

/*
树状数组（二叉索引树，Binary Indexed Tree，BIT）

        
*/

#include <iostream>
#include <string>
using namespace std;

const int maxn = 50010;
int sum[maxn];
int a[maxn];
int n ;

int lowbit (int x)    //取x的最低位1，比如4，则返回4，如5，则返回1
{
    return x & (-x) ;
}

void add(int i,int value)    //将第i个元素增加value
{
    //i的祖先都要增加val
    while(i<=n)
    {
        sum[i] += value ;
        i += lowbit(i) ;    //将i的最低位1补1，得到其祖先

    }
}
int Sum(int i)    //求前i项和
{
    int count = 0;
    //将前i项分段
    while(i>0)
    {
        count += sum[i] ;
        i -= lowbit(i) ;    //去掉i的最低位1
    }
    return count ;
}

void main()
{
    int t ;
    cin>>t ;
    for(int cnt = 1;cnt<=t;cnt++)
    {
        memset(sum,0,sizeof(sum));
        cout<<"Case "<<cnt<<":"<<endl;
        cin>>n ;
        for(int i=1;i<=n;i++)    //注意数组从1开始的
        {
            scanf("%d",&a[i]);
            add(i,a[i]) ;
        }
        string s;
        while(cin>>s , s !="End")
        {
            int i,j;
            scanf("%d%d",&i,&j);
            if(s=="Query")
            {
                printf("%d\n",Sum(j)-Sum(i-1));
            }
            if(s=="Add")
            {
                add(i,j) ;
            }
            if(s=="Sub")
            {
                add(i,-j);
            }
        }
    }
}