首页 > 代码库 > hdu1166 敌兵布阵(线段树)

hdu1166 敌兵布阵(线段树)

2024-08-04 01:47:32   222人阅读

题目意思：

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


hdu1166 敌兵布阵(线段树)


联系
我们