HDU 1166 树状数组

 

每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令。

蓝书196页sum函数和add函数模板。

1.C数组初始化为0。

2.输入n个整数时进行1次add操作。

3.Add i j 进行add(i,j)操作。

4.Sub i j 进行add(i,-j)操作。

5.Query 输出sum(j)-sum(i-1)。

#include<stdio.h>
#include<string.h>
int n,a[50005],c[50005];
int lowbit(int x)
{
	return x&(-x);
}
int sum(int x)
{
	int ret=0;
	while(x>0)
	{
		ret+=c[x];
		x-=lowbit(x);
	}
	return ret;
}
int add(int x,int d)
{
	while(x<=n)
	{
		c[x]+=d;
		x+=lowbit(x);
	}
}
int main()
{
  int i,j,k,t,ans1,ans2;
  char s[100];
  scanf("%d",&t);
  for(k=1;k<=t;k++)
  {
  	printf("Case %d:\n",k);
  	memset(c,0,sizeof(c));
  	scanf("%d",&n);
  	for(i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		add(i,a[i]);
	}	
  	while(scanf("%s",s)&&strcmp(s,"End")!=0)
  	{
  	 scanf("%d%d",&i,&j);
  	 if(strcmp(s,"Add")==0) 
	 {
	   add(i,j);	
	 }
	 else if(strcmp(s,"Sub")==0)	
	 {
	   add(i,-j);	
	 }
	 else
	 {
	   ans1=sum(i-1);
	   ans2=sum(j);
	   ans2-=ans1;
	   printf("%d\n",ans2);	
	 }
	}
  }
}