2013暑假江西联合训练赛（赛后） 二进制求和 (树状数组，这里需要变形下)

Problem Description

平时我们经常会将一个数用二进制的形式来表示，我们用 BSUM（X） 表示X用二进制形式表示中1的个数。现在给你一个长度为n (n<=100000)的数组g (0<=g[i]<10^9)。 有q(q<=100000)次操作
1 a b (1<=a<=b<=n) 表示要输出 BSUM[a]+BSUM[a+1]... BSUM[b] 的和。
2 a b (1<=a<=n,0<=b<=10^9) 表示将g[a]的值变成b,即g[a]=b
3 a b (1<=a<=b<=n) 表示g[i]=2*g[i] (a<=i<=b),即将数组从a到b的值全部都乘以2.

Input

第一行输入T（<=100），代表测试组数
每组测试先输入一行n，下一行输入n个数a[i]（1<=i<=n）.
然后输入q表示询问次数
接下来q行代表每次询问。

Sample Input

2
5
1 2 3 4 5
3
1 2 4
2 3 0
1 1 5
5
1 4 8 9 10
5
1 1 5
2 1 0
3 1 5
1 2 2
1 1 5

Sample Output

4
5
7
1
6

题目大意：很明显的树状数组、但需要小小的变形。

思路见代码咯

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAX 100010
using namespace std;
int c[MAX];
int b[MAX];
int BitCount(int n)         //计算二进制1的个数
{
    int c=0;
    for(c=0;n;++c)
    {
        n&=(n-1);
    }
    return c;
}
int lowbit(int x)
{
    return x&(-x);
}
void updata(int x,int d)
{
    while(x<=MAX)
    {
        c[x]=c[x]+d;
        x=x+lowbit(x);
    }
}
int sum(int x)
{
    int res=0;
    while(x>0)
    {
        res=res+c[x];
        x=x-lowbit(x);
    }
    return res;
}
int main()
{
    int t;
    int n,i;
    int x,y;
    int p,q;
    int ans;
    int s,s1,s2;
    scanf("%d",&t);
    while(t--)
    {
        ans=0;
        memset(b,0,sizeof(b));
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&b[i]);
            s=BitCount(b[i]);
            updata(i,s);
        }
        scanf("%d",&p);
        while(p--)
        {
            scanf("%d",&q);
            if(q==1)
            {
                scanf("%d%d",&x,&y);
                ans=sum(y)-sum(x-1);
                printf("%d\n",ans);
            }
            else if(q==2)
            {
                scanf("%d%d",&x,&y);
                s1=BitCount(y);         //统计y中1的个数
                s2=BitCount(b[x]);      //统计b[x]中 1 的个数
                updata(x,-s2);          //消去原有b[x] 中 1的个数
                updata(x,s1);           //修改b[x] 中 1 的个数
                b[x]=y;                 //一定要记住修改b[x]的值，WA了4次
            }
            else if(q==3)
            {
                scanf("%d%d",&x,&y);    //乘以2 1的个数不变
            }
        }
    }
    return 0;
}