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

HDU 1166 敌兵布阵(线段树)

2024-07-16 00:32:17   223人阅读

A - 敌兵布阵
Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u
Submit Status

Description

C国的死对头A国这段时间正在进行军事演习，所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段，所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动，可能增加或减少若干人手,但这些都逃不过C国的监视。 
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动，而Derek每次询问的段都不一样，所以Tidy不得不每次都一个一个营地的去数，很快就精疲力尽了，Derek对Tidy的计算速度越来越不满:"你个死肥仔，算得这么慢，我炒你鱿鱼!”Tidy想：“你自己来算算看，这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下，Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说：“死肥仔，叫你平时做多点acm题和看多点算法书，现在尝到苦果了吧!”Tidy说："我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼，这么算他真的会崩溃的，聪明的读者，你能写个程序帮他完成这项工作吗？不过如果你的程序效率不够高的话，Tidy还是会受到Derek的责骂的. 
 

Input

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

Output

对第i组数据,首先输出“Case i:”和回车, 
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。 
 

Sample Input

1101 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
 

Sample Output

Case 1:
6
33
59
 
 
线段树的入门题，从这题可以掌握怎么去创建线段树，更新线段树和查询线段树
 1 #include<cstdio> 2 #include<cstring> 3 #include<string> 4 #include<stdlib.h> 5 #include<iostream> 6 #include<algorithm> 7 using namespace std; 8 struct node                             //节点结构体 9 {10     int l,r;11     int num;12 }a[250000];13 int ans;14 void contree(int l,int r,int now)       //创建线段树15 {16     a[now].l=l;17     a[now].r=r;18     if(l==r)                            //如果l==r,说明没有子节点了,那么此时为单个营地,输入人数19     {20         scanf("%d",&a[now].num);21         return ;22     }23     int mid=(a[now].l+a[now].r)/2;      24     contree(l,mid,now<<1);              //递归左子树25     contree(mid+1,r,now<<1|1);          //递归右子树26     a[now].num=a[now<<1].num+a[now<<1|1].num;27 }                                        28 void addtree(int l,int r,int now,int vis,int val)   //更新线段树29 {30     if(l==r&&l==vis)31     {32         a[now].num+=val;33         return ;34     }35     int mid=(a[now].l+a[now].r)/2;36     if(vis>mid)37         addtree(mid+1,r,now<<1|1,vis,val);38     if(vis<=mid)39         addtree(l,mid,now<<1,vis,val);40     a[now].num=a[now<<1].num+a[now<<1|1].num;        //每次更改某个值的时候,被查找过的数的值也要更新,这点很重要41 }42 void sumtree(int l,int r,int now,int L,int R)        //查询线段树43 {44     if(L<=l&&r<=R)45     {46         ans+=a[now].num;47         return ;48     }49     int mid=(a[now].l+a[now].r)/2;50     if(L>mid)51         sumtree(mid+1,r,now<<1|1,L,R);52     else if(R<=mid)53         sumtree(l,mid,now<<1,L,R);54     else55     {56         sumtree(l,mid,now<<1,L,R);57         sumtree(mid+1,r,now<<1|1,L,R);58     }59 }60 int main()61 {62     int kase,n,cnt=0;63     scanf("%d",&kase);64     while(kase--)65     {66         scanf("%d",&n);67         contree(1,n,1);68         printf("Case %d:\n",++cnt);69         while(1)70         {71             getchar();72             string str;73             cin>>str;74             int x,y;75             ans=0;76             if(str=="End")77                 break;78             scanf("%d %d",&x,&y);79             if(str=="Query")80             {81                 sumtree(1,n,1,x,y);82                 printf("%d\n",ans);83             }84             else if(str=="Add")85                 addtree(1,n,1,x,y);86             else87                 addtree(1,n,1,x,-y);88         }89     }90     return 0;91 }
View Code

 

 


联系
我们