hdu1166 敌兵布阵(线段树)

题目意思：

http://acm.hdu.edu.cn/showproblem.php?pid=1166


第一行一个整数T，表示有T组数据。

每组数据第一行一个正整数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条命令

题目分析：

线段树的应用，自己的第一个线段树代码，做题之前一直看高级数据结构之县单数并参考了大神代码，自己认为，线段树就是二叉树，只是结点的信息量比较大，主要的是怎么保存结点的信息，这个要根据题目意思进行修改，例如本题要求的数总人数，那么需要保存i-j的总人数，代码如下

AC代码：

/**
*线段树的简单应用,用scanf和printf输入输出
*cin和cout会超时
*/
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int a[50001];//存储结点的位置
struct node{
int left,right;//线段区间
int sum;//区间上的人数
}b[150005];
void Build(int left,int right,int i){//建树
int mid;
b[i].left=left;
b[i].right=right;
if(left==right){//先用数组存下该队的人
b[i].sum=a[left];
return ;
}
//cout<<"ok"<<endl;
mid=(left+right)/2;
Build(left,mid,2*i);//递归建立左子树
Build(mid+1,right,2*i+1);//递归建立右子树
b[i].sum=b[2*i].sum+b[2*i+1].sum;//记录该结点左右子树的值
}

void Add(int j,int num,int i){
if(b[i].left==b[i].right){//到达叶子结点
b[i].sum+=num;
return ;//注意返回
}
else{
b[i].sum+=num;//他的所有父节点都增加
if(j<=b[i*2].right) Add(j,num,2*i);//递归左子树
else Add(j,num,2*i+1);//建立右子树
}
}

int Query(int left,int right,int i){//查询
int mid;
if(b[i].left==left&&b[i].right==right) return b[i].sum;
mid=(b[i].left+b[i].right)/2;
if(right<=mid) return Query(left,right,2*i);//在左子树
else if(left>mid) return Query(left,right,2*i+1);//在右子树
else return Query(left,mid,2*i)+Query(mid+1,right,2*i+1);//否则在中间
}
int main()
{
int t,k=0,i,j,num,n;
char s[10];
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
}
//cout<<"Case "<<++k<<":"<<endl;
printf("Case %d:\n",++k);
Build(1,n,1);//建树
while(1){
scanf("%s",s);
if(!strcmp(s,"End")) break;
scanf("%d%d",&j,&num);
if(!strcmp(s,"Query")){
printf("%d\n",Query(j,num,1));
//cout<<Query(j,num,1)<<endl;
}
if(!strcmp(s,"Add")) Add(j,num,1);//从根结点开始
if(!strcmp(s,"Sub")) Add(j,-num,1);

}
}
return 0;
}