HDU 1166 敌兵布阵（树状数组+单点修改+区间查询）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

题意

        有n个工兵营地，每个营地初始有初始的人数，接下来有四种操作：

1. Add i j,i和j为正整数,表示第i个营地增加j个人
2. Sub i j ,i和j为正整数,表示第i个营地减少j个人
3. Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数
4. End 表示结束，这条命令在每组数据最后出现

        每进行一次Query输出对应区间内的人数。

思路

        采用树状数组的结构存储数据，树状数组的精髓就在于lowbit函数。单点修改只需将当前结点的所有父节点进行更新，即用lowbit函数值向上更新数据。区间查询时，因为树状数组求和是求从第一个位置开始的，所以运用前缀和的思想就可以得到任意一个区间和。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn=5e4+10;

int c[maxn];
//取出x二进制的最低位1
int lowbit(int x)
{
	return x&(-x);
}
//单点更新
int update(int x,int y,int n)
{
	for(int i=x;i<=n;i+=lowbit(i)) c[i]+=y;
}
//区间查询1到x的和
int getsum(int x)
{
	int ans=0;
	for(int i=x;i>0;i-=lowbit(i)) ans+=c[i];
	return ans;
}

int main()
{
//	freopen ("input.txt" , "r" , stdin);
//	freopen ("output.txt" , "w" , stdout);
	int t,Case=1;
	scanf("%d",&t);
	while(t--){
		int n,z;
		scanf("%d",&n);
		memset(c,0,sizeof(c));
		for(int i=1;i<=n;i++) scanf("%d",&z),update(i,z,n);
		printf("Case %d:\n",Case++);
		while(1){
			int x,y;
			string s;
			cin>>s;
			if(s=="End") break;
			scanf("%d %d",&x,&y);
			if(s=="Add") update(x,y,n);
			else if(s=="Sub") update(x,-y,n);
			else if(s=="Query") printf("%d\n",getsum(y)-getsum(x-1));//前缀和思想求区间和
		}
	}
	return 0;
}