HDU 1166 树状数组

最新推荐文章于 2019-09-06 21:56:29 发布

敲代码的欧文

最新推荐文章于 2019-09-06 21:56:29 发布

阅读量249

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/Irving0323/article/details/83065618

本文深入解析了一种处理敌军工兵营地数据的算法，通过使用树状数组实现快速更新和查询操作。具体介绍了如何通过Add和Sub命令来增加或减少特定营地的人数，以及如何查询指定范围内营地的总人数。文章提供了详细的C语言实现代码，包括关键的sum和add函数模板。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

每组数据第一行一个正整数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);	
	 }
	}
  }
}