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