HDU1166 敌兵布阵 —— 线段树单点修改

Posted 2020-10-11 h_z_cong

题目链接：https://vjudge.net/problem/HDU-1166

 

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

 

代码一：

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <cmath>
 5 #include <algorithm>
 6 #include <vector>
 7 #include <queue>
 8 #include <stack>
 9 #include <map>
10 #include <string>
11 #include <set>
12 using namespace std;
13 typedef long long LL;
14 const double EPS = 1e-8;
15 const int INF = 2e9;
16 const LL LNF = 2e18;
17 const int MAXN = 5e4+10;
18 
19 int sum[MAXN*4];
20 
21 void push_up(int u)
22 {
23     sum[u] = sum[u*2] + sum[u*2+1];
24 }
25 
26 void build(int u, int l, int r)
27 {
28     if(l==r)
29     {
30         scanf("%d", &sum[u]);
31         return;
32     }
33     int mid = (l+r)/2;
34     build(u*2, l, mid);
35     build(u*2+1, mid+1, r);
36     push_up(u);
37 }
38 
39 void add(int u, int l, int r, int pos, int val)
40 {
41     if(l==r)
42     {
43         sum[u] += val;
44         if(sum[u]<0) sum[u] = 0;
45         return;
46     }
47 
48     int mid = (l+r)/2;
49     if(pos<=mid) add(u*2, l, mid, pos, val);
50     else add(u*2+1, mid+1, r, pos, val);
51     push_up(u);
52 }
53 
54 int query(int u, int l, int r, int x, int y)
55 {
56     if(x<=l && r<=y)
57         return sum[u];
58 
59     int ret = 0;
60     int mid = (l+r)/2;
61     if(x<=mid) ret += query(u*2, l, mid, x, y);
62     if(y>=mid+1) ret += query(u*2+1, mid+1, r, x, y);
63     return ret;
64 }
65 
66 int main()
67 {
68     int T, n;
69     scanf("%d", &T);
70     for(int kase = 1; kase<=T; kase++)
71     {
72         scanf("%d", &n);
73         build(1, 1, n);
74         printf("Case %d:\n", kase);
75         char op[10]; int a, b;
76         while(scanf("%s", op) && strcmp(op, "End"))
77         {
78             scanf("%d%d", &a, &b);
79             if(op[0]==‘A‘) add(1, 1, n, a, b);
80             else if(op[0]==‘S‘) add(1, 1, n, a, -b);
81             else printf("%d\n", query(1, 1, n, a, b));
82         }
83     }
84 }

View Code

 

代码二：

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <cmath>
 5 #include <algorithm>
 6 #include <vector>
 7 #include <queue>
 8 #include <stack>
 9 #include <map>
10 #include <string>
11 #include <set>
12 using namespace std;
13 typedef long long LL;
14 const double EPS = 1e-8;
15 const int INF = 2e9;
16 const LL LNF = 2e18;
17 const int MAXN = 5e4+10;
18 
19 int sum[MAXN*4];
20 
21 void push_up(int u)
22 {
23     sum[u] = sum[u*2] + sum[u*2+1];
24 }
25 
26 void add(int u, int l, int r, int pos, int val)
27 {
28     if(l==r)
29     {
30         sum[u] += val;
31         if(sum[u]<0) sum[u] = 0;
32         return;
33     }
34 
35     int mid = (l+r)/2;
36     if(pos<=mid) add(u*2, l, mid, pos, val);
37     else add(u*2+1, mid+1, r, pos, val);
38     push_up(u);
39 }
40 
41 int query(int u, int l, int r, int x, int y)
42 {
43     if(x<=l && r<=y)
44         return sum[u];
45 
46     int ret = 0;
47     int mid = (l+r)/2;
48     if(x<=mid) ret += query(u*2, l, mid, x, y);
49     if(y>=mid+1) ret += query(u*2+1, mid+1, r, x, y);
50     return ret;
51 }
52 
53 int main()
54 {
55     int T, n;
56     scanf("%d", &T);
57     for(int kase = 1; kase<=T; kase++)
58     {
59         scanf("%d", &n);
60         char op[10]; int a, b;
61         memset(sum, 0, sizeof(sum));
62         for(int i = 1; i<=n; i++)
63         {
64             scanf("%d", &a);
65             add(1, 1, n, i, a);
66         }
67         printf("Case %d:\n", kase);
68 
69         while(scanf("%s", op) && strcmp(op, "End"))
70         {
71             scanf("%d%d", &a, &b);
72             if(op[0]==‘A‘) add(1, 1, n, a, b);
73             else if(op[0]==‘S‘) add(1, 1, n, a, -b);
74             else printf("%d\n", query(1, 1, n, a, b));
75         }
76     }
77 }

View Code