-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathlazy_segtree_monoid.cpp
124 lines (112 loc) · 3.38 KB
/
lazy_segtree_monoid.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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
// hoge
#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using vi = vector<i64>;
using vvi = vector<vi>;
template <class T, class T2>
class LazySegTree {
using F = function<T(T, T)>;
using G = function<T(T, T2)>;
using H = function<T2(T2, T2)>;
using P = function<T2(T2, int)>;
int n;
vector<T> data;
vector<T2> lazy;
const F f;
const G g;
const H h;
const P p;
const T e;
const T2 e2;
public:
LazySegTree(const vector<T>& as, const F f, const G g, const H h, const P p, const T& e, const T2 e2) : f(f), g(g), h(h), p(p), e(e), e2(e2) {
n = 1;
while (n < as.size()) n <<= 1;
data.assign(2 * n, e);
lazy.assign(2 * n, e2);
for (int i = 0; i < as.size(); i++) {
data[n + i] = as[i];
}
for (int i = n - 1; i > 0; i--) {
data[i] = f(data[2 * i], data[2 * i + 1]);
}
}
void propagate(int k, int len) {
if (lazy[k] != e2) {
if (k < n) {
lazy[2 * k] = h(lazy[2 * k], lazy[k]);
lazy[2 * k + 1] = h(lazy[2 * k + 1], lazy[k]);
}
data[k] = g(data[k], p(lazy[k], len));
lazy[k] = e2;
}
}
T update(int a, int b, const T2& x, int k, int l, int r) {
propagate(k, r - l);
if (r <= a || b <= l) {
return data[k];
} else if (a <= l && r <= b) {
lazy[k] = h(lazy[k], x);
propagate(k, r - l);
return data[k];
} else {
return data[k] = f(update(a, b, x, 2 * k, l, (l + r) >> 1), update(a, b, x, 2 * k + 1, (l + r) >> 1, r));
}
}
T update(int a, int b, const T2& x) {
return update(a, b, x, 1, 0, n);
}
T query(int a, int b, int k, int l, int r) {
propagate(k, r - l);
if (r <= a || b <= l) {
return e;
} else if (a <= l && r <= b) {
return data[k];
} else {
return f(query(a, b, 2 * k, l, (l + r) >> 1), query(a, b, 2 * k + 1, (l + r) >> 1, r));
}
}
T query(int a, int b) {
return query(a, b, 1, 0, n);
}
};
int main() {
int n;
cin >> n;
vi as(n);
// RUQ, RMQ
// auto f = [](i64 a, i64 b) {return min(a, b);};
// auto g = [](i64 a, i64 b) {return b < 0 ? a : b;};
// auto h = [](i64 a, i64 b) {return b < 0 ? a : b;};
// auto p = [](i64 a, int b) {return a;};
// LazySegTree<i64, i64> tree(as, f, g, h, p, 1e18, -1);
// RAQ, RMQ
// auto f = [](i64 a, i64 b) {return min(a, b);};
// auto g = [](i64 a, i64 b) {return a + b;};
// auto h = [](i64 a, i64 b) {return a + b;};
// auto p = [](i64 a, i64 b) {return a;};
// LazySegTree<i64, i64> tree(as, f, g, h, p, 1e18, 0);
// RUQ, RSQ
const i64 INF = 1e18;
auto f = [](i64 a, i64 b) {return a + b;};
auto g = [](i64 a, i64 b) {return b == INF ? a : b;};
auto h = [](i64 a, i64 b) {return b == INF ? a : b;};
auto p = [](i64 a, i64 b) {return a == INF ? INF : a * b;};
LazySegTree<i64, i64> tree(as, f, g, h, p, 0, INF);
int q;
cin >> q;
while (q--) {
int c;
cin >> c;
if (c == 0) {
int s, t, x;
cin >> s >> t >> x;
tree.update(s, t + 1, x);
} else {
int s, t;
cin >> s >> t;
cout << tree.query(s, t + 1) << endl;
}
}
}