树状数组 HDU1166

          继续上次那题,我们现在采用树状数组来解。树状数组主要用来进行区间求和统计和定点更新操作,树状数组的存储是如下方式的:假设数组a为a[1]、a[2]、a[3].......a[n],现在我们另数组c[1]=a[1], c[2]=a[1]+a[2], c[3]=a[3], c[4]=a[1]+a[2]+a[3]+a[4]......,这是啥结构呢?其实c[i]=a[i-2^n+1]+......+a[i],其中n为 i 所对应的二进制数末尾0的够数。很容易得知,这种结构后,区间求和将变为log(n)级别。我们假设一个数为x,那么利用x&(-x)可以得出x的2^n的值。于是我们进行节点更新时,可以利用这个技巧进行更新及统计。相关代码如下:
#include <cstdio>
#include<cstring>
using namespace std;
#define MAXN 50010
int c[MAXN],n;
int lowBit(int x)
{
    return x&(-x);
}
void update(int i,int value)
{
    while( i <= n )
    {
        c[i] += value;
        i += lowBit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s += c[i];
        i -= lowBit(i);
    }
    return s;
}
int main()
{
    int a,b,i,j,T,cnt;
    char str[20];
    scanf("%d",&T);
    i=1;
    while(T--)
    {
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(j = 1; j <= n; j++)
        {
            scanf("%d",&cnt);
            update(j,cnt);

        }
        printf("Case %d:\n",i);
        i++;
        while(scanf("%s",str))
        {
            if(str[0]=='E')
                break;
             scanf("%d %d",&a,&b);
            if(str[0]=='A')      update(a,b);
               else if(str[0]=='S')    update(a,-b);
                 else
                    printf("%d\n",(sum(b)-sum(a-1)));
        }
    }
    return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值