线段树掌握的很差，打算从头从最简单的开始刷一波， 嗯。。就从这个题开始吧！
1 #include <iostream>
2 #include <cstdio>
3 #include <cstring>
4 #include <queue>
5 #include <cstdlib>
6 #include <algorithm>
7 const int maxn = 50000+10;
8 using namespace std;
9 int a[maxn], n;
10 struct line
11 {
12 int l, r, val;
13 }tr[maxn<<2];
14
15 void build(int o, int l, int r)
16 {
17 tr[o].l = l; tr[o].r = r;
18 if(l == r)
19 {
20 tr[o].val = a[l];
21 return;
22 }
23 int mid = (l+r)>>1;
24 build(2*o, l, mid);
25 build(2*o+1, mid+1, r);
26 tr[o].val = tr[2*o].val+tr[2*o+1].val;
27 }
28 int query(int o, int l, int r)
29 {
30 if(tr[o].l==l && tr[o].r==r)
31 return tr[o].val;
32 int mid = (tr[o].l+tr[o].r)>>1;
33 if(r <= mid) query(2*o, l, r);
34 else if(l > mid) query(2*o+1, l, r);
35 else
36 return (query(2*o, l, mid)+query(2*o+1, mid+1, r));
37 }
38 void update(int o, int p, int add)
39 {
40 if(tr[o].l==tr[o].r&&tr[o].l==p)
41 {
42 tr[o].val += add;
43 return;
44 }
45 int mid = (tr[o].l+tr[o].r)>>1;
46 if(p<=mid) update(2*o, p, add);
47 else update(2*o+1, p, add);
48 tr[o].val = tr[2*o].val+tr[2*o+1].val;
49 }
50 int main()
51 {
52 int t, i, ca = 1;
53 int p, add, l, r;
54 char s[20];
55 scanf("%d", &t);
56 while(t--)
57 {
58 scanf("%d", &n);
59 for(i = 1; i <= n; i++)
60 scanf("%d", &a[i]);
61 printf("Case %d:\n", ca++);
62
63 build(1, 1, n);
64 while(~scanf("%s", s))
65 {
66 if(strcmp(s, "End")==0) break;
67 if(s[0]==‘Q‘)
68 {
69 scanf("%d%d", &l, &r);
70 printf("%d\n", query(1, l, r));
71 }
72 if(s[0]==‘A‘)
73 {
74 scanf("%d%d", &p, &add);
75 update(1, p, add);
76 }
77 if(s[0]==‘S‘)
78 {
79 scanf("%d%d", &p, &add);
80 update(1, p, -add);
81 }
82 }
83 }
84 return 0;
85 }
注释的代码：
1 #include <iostream>
2 #include <cstdio>
3 #include <cstring>
4 #include <queue>
5 #include <cstdlib>
6 #include <algorithm>
7 const int maxn = 50000+10;
8 using namespace std;
9 int a[maxn], n;
10 struct line
11 {
12 int l, r, val; //val表示该区间的和
13 }tr[maxn<<2];
14
15 void build(int o, int l, int r) //o代表当前节点编号
16 {
17 tr[o].l = l; tr[o].r = r;
18 if(l == r)
19 {
20 tr[o].val = a[l];
21 return;
22 }
23 int mid = (l+r)>>1;
24 build(2*o, l, mid);
25 build(2*o+1, mid+1, r);
26 tr[o].val = tr[2*o].val+tr[2*o+1].val; //建树把值从下往上加起来
27 }
28 int query(int o, int l, int r) //求l到r的和
29 {
30 if(tr[o].l==l && tr[o].r==r) //节点的区间吻合返回
31 return tr[o].val;
32 int mid = (tr[o].l+tr[o].r)>>1;
33 if(r <= mid) query(2*o, l, r);
34 else if(l > mid) query(2*o+1, l, r);
35 else
36 return (query(2*o, l, mid)+query(2*o+1, mid+1, r)); //横跨了区间
37 }
38 void update(int o, int p, int add) //对p节点增加add
39 {
40 if(tr[o].l==tr[o].r&&tr[o].l==p)
41 {
42 tr[o].val += add;
43 return;
44 }
45 int mid = (tr[o].l+tr[o].r)>>1;
46 if(p<=mid) update(2*o, p, add);
47 else update(2*o+1, p, add);
48 tr[o].val = tr[2*o].val+tr[2*o+1].val; //找到值以后的更新
49 }
50 int main()
51 {
52 int t, i, ca = 1;
53 int p, add, l, r;
54 char s[20];
55 scanf("%d", &t);
56 while(t--)
57 {
58 scanf("%d", &n);
59 for(i = 1; i <= n; i++)
60 scanf("%d", &a[i]);
61 printf("Case %d:\n", ca++);
62
63 build(1, 1, n);
64 while(~scanf("%s", s))
65 {
66 if(strcmp(s, "End")==0) break;
67 if(s[0]==‘Q‘)
68 {
69 scanf("%d%d", &l, &r);
70 printf("%d\n", query(1, l, r));
71 }
72 if(s[0]==‘A‘)
73 {
74 scanf("%d%d", &p, &add);
75 update(1, p, add);
76 }
77 if(s[0]==‘S‘)
78 {
79 scanf("%d%d", &p, &add);
80 update(1, p, -add);
81 }
82 }
83 }
84 return 0;
85 }
HDU 1166 敌兵布阵 （线段树 单点更新）
时间： 2024-08-24 04:13:52