树状数组的基本作用就是
- 给一段数组,单点更新,成段求和。
- 还有一些题是按照给出的结点顺序(或者需要按照某种需要排序)遍历,分别求出比当前结点的值小的结点数(有时也要把这些值的和求出来)。
需要注意的一点就是树状数组处理的不管是数组还是线段都是从1开始计数的,所以有些题目需要加1操作。
HDU1166 地兵布阵
此题是赤裸裸的树状数组,当然用线段树也是可以做的。
单点更新,成段求和。
//31MS 604K
#include <stdio.h>
#include <memory.h>
#define lowbit(x) ((x)&(-x))
#define M 50004
int n,a[M],c[M];
void update(int p,int d)
{
while(p<=n){
c[p]+=d;
p+=lowbit(p);
}
}
int sum(int p)
{
int ans=0;
while(p>0){
ans+=c[p];
p-=lowbit(p);
}
return ans;
}
int main()
{
int T,x,y,i,ncase=0;
char op[12];
scanf("%d",&T);
while(T--)
{
memset(c,0,sizeof(c));
printf("Case %d:\n",++ncase);
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
update(i,a[i]);
}
while(scanf("%s",op) && op[0]!='E')
{
scanf("%d%d",&x,&y);
if(op[0]=='Q') printf("%d\n",sum(y)-sum(x-1));
else
{
if(op[0]=='S') y=-y;
update(x,y);
}
}
}
return 0;
}