标签：树状数组   数据结构   

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

敌兵布阵

1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End 

Case 1:
6
33
59

直接用数组做，添加删除的效率非常高，但是查询操作的时间复杂度，40000条命令 X  最多50000个营地 显然会超时

因而这里运用了树状数组这个巧妙的数据结构

直接上代码吧，原理以后再整理。。

【源代码】

#include<iostream>
#include<string>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn = 50050;
int num[maxn];
int s[maxn];
int n;
int lowbit(int x){ //求x二进制表示的最后一个1所在位置，，当然返回的值是最后一个1及其后面0所组成的二进制数
	return x&(-x);
}
void modify(int a,int b,int ord){
	if(ord){
		for(int i=a;i<=n;i+=lowbit(i)) //通过加 lowbit 来实现找到父亲节点
			s[i]+=b; //add
	}
	else{
		for(int i=a;i<=n;i+=lowbit(i)) /sub
			s[i]-=b;
	}
}
int sum(int x){
	int ans = 0;
	for(int i=x;i>0;i-=lowbit(i)) //通过 减lowbit 实现向左边节点的移动 //直到求得所有的值相加
		ans+=s[i];            //这样相加的次数明显比一个个遍历相加块的多，因为每个s[i]储存的都是几个num[]的和
         return ans;
}
void solve(int a,int b){
	printf("%d\n",sum(b)-sum(a-1)); //求得a - b区间的和
}
int main(){
	int T,cas=0;
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		for(int i=1;i<=n;i++){
			scanf("%d",&num[i]);
		}
		cout<<"Case "<<++cas<<":"<<endl;
		memset(s,0,sizeof(s));
		for(int i=1;i<=n;i++){
			for(int j=i-lowbit(i)+1;j<=i;j++){
				if(j<1) continue;
				s[i]+=num[j]; //初始化，构造s
                }
		}
		char cmd[30];
		int order1,order2;
		while(true){
			scanf(" %s",cmd);
			if(strcmp(cmd,"End")==0)
				break;
			scanf("%d%d",&order1,&order2);
			if(strcmp(cmd,"Add")==0)
			{
				modify(order1,order2,1);
			}
			else if(strcmp(cmd,"Sub")==0){
				modify(order1,order2,0);
			}
			else{
				solve(order1,order2);
			}
		}
		
	}
	return 0;
}

HDU 1166 敌兵布阵 （树状数组）

标签：树状数组   数据结构   

原文地址：http://blog.csdn.net/chaiwenjun000/article/details/47761503