hdu 1166 敌兵布阵（线段树）

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

Problem 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
1 10 1 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
 
分析：此题为线段树的标准模板题，坑爹的是用暴力也能AC，不过为了学习线段树，还是用线段树写一下
 
代码：
 1 #include<iostream>
 2 using namespace std;
 3 int ar[50010];
 4 int sum[200010];
 5 void buildingsum(int p,int s,int e)
 6 {
 7     int m=(s+e)/2;
 8     if(s==e)
 9     {
10         sum[p]=ar[s];
11         return;
12     }
13     else
14     {
15         buildingsum(p*2,s,m);
16         buildingsum(p*2+1,m+1,e);
17         sum[p]=sum[p*2]+sum[p*2+1];
18     }
19 }
20 void add_sub(int p,int a,int b,int s,int e)
21 {
22     int m;
23     sum[p]+=b;
24     do
25     {
26         m=(s+e)/2;
27         if(a>m)
28         {
29             p=2*p+1;
30             sum[p]+=b;
31             s=m+1;
32         }
33         else
34         {
35             p=2*p;
36             sum[p]+=b;
37             e=m;
38         }
39     }while(s!=e);
40 }
41 int query(int p,int a,int b,int s,int e)
42 {
43     int m=(s+e)/2;
44     if(a==s&&e==b)
45         return sum[p];
46     else if(a>m)
47         return query(p*2+1,a,b,m+1,e);
48     else if(b<=m)
49         return query(p*2,a,b,s,m);
50     else
51         return query(p*2,a,m,s,m)+query(p*2+1,m+1,b,m+1,e);
52 }
53 int main()
54 {
55     int n,t,k=1;
56     char str[100];
57     scanf("%d",&t);
58     while(t--)
59     {
60         int i;
61         scanf("%d",&n);
62         for(i=1;i<=n;i++)
63             scanf("%d",&ar[i]);
64         ar[i]=0;
65         int a,b;
66         printf("Case %d:\n",k++);
67         memset(sum,0,sizeof(sum));
68         buildingsum(1,1,n);
69         while(~scanf("%s",str))
70         {
71             if(str[0]=='E')
72                 break;
73             scanf("%d%d",&a,&b);
74             if(str[0]=='Q')
75                 printf("%d\n",query(1,a,b,1,n));
76             else if(str[0]=='A')
77                 add_sub(1,a,b,1,n);
78             else if(str[0]=='S')
79                 add_sub(1,a,-b,1,n);    
80         }
81     }
82     return 0;
83 }

 一年之后，再做这道题

代码有什么不一样么？： 2013-07-29

 1 #include<iostream>
 2 #include<stdio.h>
 3 #include<string.h>
 4 using namespace std;
 5 #define N 50010
 6 
 7 struct Nod
 8 {
 9     int l,r;
10     int sum;
11 }node[N<<2];
12 
13 void buildingTree(int l,int r,int p)
14 {
15     int mid = (l+r)>>1;
16     node[p].l = l;
17     node[p].r = r;
18     if(l==r)
19     {
20         scanf("%d",&node[p].sum);
21         return;
22     }
23     buildingTree(l,mid,p*2);
24     buildingTree(mid+1,r,p*2+1);
25     node[p].sum = node[p*2].sum + node[p*2+1].sum;
26 }
27 
28 int queryTree(int a,int b,int p)
29 {
30     int mid = (node[p].l+node[p].r)>>1;
31     if(node[p].l==a&&node[p].r==b)  return node[p].sum;
32     if(mid>=b)  return queryTree(a,b,p*2);
33     else if(mid<a)  return queryTree(a,b,p*2+1);
34     else    return queryTree(a,mid,p*2)+queryTree(mid+1,b,p*2+1);
35 }
36 
37 void addTree(int a,int b,int p)
38 {
39     int mid = (node[p].l + node[p].r)>>1;
40     if(node[p].l==node[p].r&&node[p].l==a)
41     {
42         node[p].sum+=b;
43         return;
44     }
45     if(mid>=a)  addTree(a,b,p*2);
46     else if(mid<a)  addTree(a,b,p*2+1);
47     node[p].sum+=b;
48 }
49 
50 int main()
51 {
52     int t,cnt=1;
53     scanf("%d",&t);
54     while(t--)
55     {
56         int n;
57         scanf("%d",&n);
58         buildingTree(1,n,1);
59         char cmd[10];
60         int a,b;
61         printf("Case %d:\n",cnt++);
62         while(1)
63         {
64             scanf("%s",cmd);
65             if(cmd[0]=='E') break;
66             else if(cmd[0]=='Q')
67             {
68                 scanf("%d%d",&a,&b);
69                 printf("%d\n",queryTree(a,b,1));
70             }
71             else if(cmd[0]=='A')
72             {
73                 scanf("%d%d",&a,&b);
74                 addTree(a,b,1);
75             }
76             else if(cmd[0]=='S')
77             {
78                 scanf("%d%d",&a,&b);
79                 addTree(a,-b,1);
80             }
81         }
82     }
83     return 0;
84 }

 

posted @ 2012-08-30 18:53  crazy_apple  阅读(212)  评论(0)  编辑  收藏  举报