-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathlazy_segtree.cpp
101 lines (91 loc) · 1.97 KB
/
lazy_segtree.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// hoge
#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using vi = vector<i64>;
class LazySegTree {
int n;
vi node, lazy;
public:
LazySegTree(vi v) {
int s = int(v.size());
n = 1;
while (n < s) {
n <<= 1;
}
node.resize(2 * n - 1);
lazy.resize(2 * n - 1 , 0);
for (int i = 0; i < s; i++) {
node[i + n - 1] = v[i];
}
for (int i = n - 2; i >= 0; i--) {
node[i] = node[i * 2 + 1] + node[i * 2 + 2];
}
}
void eval(int k, int l, int r) {
if (lazy[k]) {
node[k] += lazy[k];
if (r - l > 1) {
lazy[2 * k + 1] += lazy[k] / 2;
lazy[2 * k + 2] += lazy[k] / 2;
}
lazy[k] = 0;
}
}
void add(int a, int b, i64 x, int k = 0, int l = 0, int r = -1) {
if (r < 0) r = n;
eval(k, l, r);
if (b <= l || r <= a) {
return;
}
if (a <= l && r <= b) {
lazy[k] += (r - l) * x;
eval(k, l, r);
} else {
add(a, b, x, 2 * k + 1, l, (l + r) / 2);
add(a, b, x, 2 * k + 2, (l + r) / 2, r);
node[k] = node[2 * k + 1] + node[2 * k + 2];
}
}
i64 rangeSum(int a, int b, int k = 0, int l = 0, int r = -1) {
if (r < 0) r = n;
eval(k, l, r);
if (b <= l || r <= a) {
return 0;
}
if (a <= l && r <= b) {
return node[k];
}
i64 vl = rangeSum(a, b, 2 * k + 1, l, (l + r) / 2);
i64 vr = rangeSum(a, b, 2 * k + 2, (l + r) / 2, r);
return vl + vr;
}
void set(int a, i64 x) {
add(a, a + 1, -get(a));
add(a, a + 1, x);
}
i64 get(int a) {
return rangeSum(a, a + 1);
}
};
int main() {
int n;
cin >> n;
vi as(n);
LazySegTree seg(as);
int q;
cin >> q;
for (int i = 0; i < q; i++) {
char c;
cin >> c;
if (c == 'A') {
int id, x;
cin >> id >> x;
seg.set(id - 1, x);
} else {
int l, r;
cin >> l >> r;
cout << seg.rangeSum(l - 1, r) << endl;
}
}
}