码迷,mamicode.com
首页 > 其他好文 > 详细

hdu1166 敌兵布阵(线段树)

时间:2014-11-24 13:37:09      阅读:188      评论:0      收藏:0      [点我收藏+]

标签:hdu   数据结构   

题目意思:

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 敌兵布阵(线段树)

标签:hdu   数据结构   

原文地址:http://blog.csdn.net/fool_ran/article/details/41442779

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!