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;}