-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlazy_segment_tree.cpp
More file actions
54 lines (47 loc) · 1.31 KB
/
lazy_segment_tree.cpp
File metadata and controls
54 lines (47 loc) · 1.31 KB
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
struct LazySegmentTree {
const int nothing = -1e9;
vector<int> val, lazy;
int n;
constexpr int f(int a, int b) { return max(a, b); }
inline void pushdown(int cur) {
if (lazy[cur])
val[cur] = lazy[cur];
if (cur < n && lazy[cur]) {
lazy[cur << 1] = lazy[cur];
lazy[cur << 1 | 1] = lazy[cur];
}
lazy[cur] = 0;
}
void update(int l, int r, int value, int cur = 1, int ll = 1, int rr = 1e9) {
rr = min(rr, n);
pushdown(cur);
if (l > r)
return;
if (l == ll && r == rr) {
lazy[cur] = value;
pushdown(cur);
return;
}
int mid = (ll + rr) >> 1;
update(l, min(r, mid), value, cur << 1, ll, mid);
update(max(l, mid + 1), r, value, cur << 1 | 1, mid + 1, rr);
val[cur] = f(val[cur << 1], val[cur << 1 | 1]);
}
int query(int l, int r, int cur = 1, int ll = 1, int rr = 1e9) {
rr = min(rr, n);
pushdown(cur);
if (l > r)
return nothing;
if (l == ll && r == rr)
return val[cur];
int mid = (ll + rr) >> 1;
return f(query(l, min(r, mid), cur << 1, ll, mid), query(max(l, mid + 1), r, cur << 1 | 1, mid + 1, rr));
}
LazySegmentTree(vector<int> arr) {
for (n = arr.size(); n & (n - 1); n++) {}
val = vector<int>(n * 2, nothing);
lazy = vector<int>(n * 2, 0);
for (int i = n + arr.size() - 1; i; i--)
val[i] = ((i < n) ? f(val[i << 1], val[i << 1 | 1]) : arr[i - n]);
}
};