树状数组 HDU1166

来源：互联网发布：golang exit status 2 编辑：程序博客网时间：2024/05/01 23:27

继续上次那题，我们现在采用树状数组来解。树状数组主要用来进行区间求和统计和定点更新操作，树状数组的存储是如下方式的：假设数组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 50010int 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