（线段树）hdoj1166-敌兵布阵

最新推荐文章于 2018-09-05 11:09:40 发布

转载最新推荐文章于 2018-09-05 11:09:40 发布 · 95 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/quintessence/p/6392448.html

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

线段树的基础应用，init函数初始化树各个结点，add，sub函数分别处理对线段树上某叶子结点的值的更新，query函数返回查询区间的值。
参考代码

 1 #include <iostream>
 2 #include <queue>
 3 #include <cstdio>
 4 #include <cstring>
 5 #include <algorithm>
 6 using namespace std;
 7 typedef long long ll;
 8 typedef unsigned long long ull;
 9 struct node
10 {
11     int left,right,mid;
12     int num;
13 }p[500005];
14 void init(int a,int b,int n)
15 {
16     p[n].left=a;
17     p[n].right=b;
18     p[n].mid=(a+b)/2;
19     p[n].num=0;
20     if(a+1==b)
21         return;
22     init(a,(a+b)/2,2*n);
23     init((a+b)/2,b,2*n+1);
24 }
25 void add(int pos,int value,int n)
26 {
27     p[n].num+=value;
28     if(p[n].left+1==p[n].right)
29         return;
30     if(pos<p[n].mid)
31         add(pos,value,2*n);
32     else
33         add(pos,value,2*n+1);
34 }
35 void sub(int pos,int value,int n)
36 {
37     p[n].num-=value;
38     if(p[n].left+1==p[n].right)
39         return;
40     if(pos<p[n].mid)
41         sub(pos,value,2*n);
42     else
43         sub(pos,value,2*n+1);
44 }
45 int query(int a,int b,int n)
46 {
47     if(p[n].left==a&&p[n].right==b)
48         return p[n].num;
49     if(a<p[n].mid)
50     {
51         if(b<=p[n].mid)
52             return query(a,b,2*n);
53         else
54             return query(a,p[n].mid,2*n)+query(p[n].mid,b,2*n+1);
55     }
56     else
57         return query(a,b,2*n+1);
58 }
59 int t;
60 int N;
61 int s;
62 char rec[10];
63 int main()
64 {
65     scanf("%d",&t);
66     for(s=1;s<=t;s++)
67     {
68         printf("Case %d:\n",s);
69         scanf("%d",&N);
70         init(1,N+1,1);
71         int i,tem,j;
72         for(i=1;i<=N;i++)
73         {
74             scanf("%d",&tem);
75             add(i,tem,1);
76         }
77         while(scanf("%s",rec)&&rec[0]!='E')
78         {
79             if(rec[0]=='A')
80             {
81                 scanf("%d %d",&i,&j);
82                 add(i,j,1);
83             }
84             else if(rec[0]=='S')
85             {
86                 scanf("%d %d",&i,&j);
87                 sub(i,j,1);
88             }
89             else if(rec[0]=='Q')
90             {
91                 scanf("%d %d",&i,&j);
92                 printf("%d\n",query(i,j+1,1));
93             }
94         }
95     }
96     return 0;
97 }

 复习时重新写的代码

 1 #include <iostream>
 2 #include<bits/stdc++.h>
 3 #include <stack>
 4 #include <queue>
 5 #include <map>
 6 #include <set>
 7 #include <cstdio>
 8 #include <cstring>
 9 #include <algorithm>
10 #include <math.h>
11 using namespace std;
12 typedef long long ll;
13 typedef unsigned long long ull;
14 const int MAX=5e4+5;
15 int sum[MAX<<2];
16 void init(int l,int r,int k)
17 {
18     if(l==r)
19     {
20         scanf("%d",&sum[k]);
21         return;
22     }
23     int mid=l+r>>1;
24     init(l,mid,2*k);
25     init(mid+1,r,2*k+1);
26     sum[k]=sum[2*k]+sum[2*k+1];
27 }
28 void update(int lo,int num,int l,int r,int k)
29 {
30     if(l==r)
31     {
32         sum[k]+=num;
33         return;
34     }
35     int mid=l+r>>1;
36     if(lo<=mid)
37         update(lo,num,l,mid,2*k);
38     else
39         update(lo,num,mid+1,r,2*k+1);
40     sum[k]=sum[2*k]+sum[2*k+1];
41 }
42 int query(int ql,int qr,int l,int r,int k)
43 {
44     if(l>=ql&&r<=qr)
45         return sum[k];
46     int mid=l+r>>1;
47     if(qr<=mid)
48         return query(ql,qr,l,mid,2*k);
49     else if(ql>mid)
50         return query(ql,qr,mid+1,r,2*k+1);
51     else return query(ql,qr,l,mid,2*k)+query(ql,qr,mid+1,r,2*k+1);
52 }
53 int t,n;
54 char com[10];
55 int lo,num;
56 int main()
57 {
58     scanf("%d",&t);
59     for(int i=1;i<=t;i++)
60     {
61         scanf("%d",&n);
62         init(1,n,1);
63         printf("Case %d:\n",i);
64         while(~scanf("%s",com))
65         {
66             if(com[0]=='E')
67                 break;
68             else
69             {
70                 scanf("%d %d",&lo,&num);
71                 if(com[0]=='A')
72                     update(lo,num,1,n,1);
73                 else if(com[0]=='S')
74                     update(lo,-num,1,n,1);
75                 else
76                     printf("%d\n",query(lo,num,1,n,1));
77             }
78         }
79     }
80 }

转载于:https://www.cnblogs.com/quintessence/p/6392448.html