/
range_assign.test.cpp
83 lines (75 loc) · 1.57 KB
/
range_assign.test.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
#define PROBLEM "https://judge.yosupo.jp/problem/aplusb"
#include "my_template.hpp"
#include "ds/segtree/range_assignment_segtree.hpp"
#include "ds/segtree/lazy_segtree.hpp"
#include "alg/monoid/add.hpp"
#include "alg/acted_monoid/sum_assign.hpp"
#include "random/base.hpp"
struct PROB {
int N, Q;
vc<ll> INIT;
vc<tuple<int, int, int>> QUERY;
};
PROB gen(int N, int Q) {
PROB p;
p.N = N, p.Q = Q;
FOR(N) { p.INIT.eb(RNG(0, 1 << 30)); }
FOR(Q) {
int t = RNG(0, 2);
int l = RNG(0, N), r = RNG(0, N);
int x = RNG(0, 1 << 30);
if (l > r) swap(l, r);
++r;
if (t == 0) p.QUERY.eb(l, r, x);
if (t == 1) p.QUERY.eb(l, r, -1);
}
return p;
}
vi sol_1(PROB p) {
vi ANS;
Lazy_SegTree<ActedMonoid_Sum_Assign<ll, -1>> seg(p.INIT);
for (auto& [l, r, x]: p.QUERY) {
if (x == -1) {
ANS.eb(seg.prod(l, r));
} else {
seg.apply(l, r, x);
}
}
return ANS;
}
vi sol_2(PROB p) {
vi ANS;
Range_Assignment_SegTree<Monoid_Add<ll>> seg(p.INIT);
for (auto& [l, r, x]: p.QUERY) {
if (x == -1) {
ANS.eb(seg.prod(l, r));
} else {
seg.assign(l, r, x);
}
}
return ANS;
}
void test() {
int N = 1 << 22, Q = 1 << 22;
PROB p = gen(N, Q);
double a = clock();
vi A = sol_1(p);
double b = clock();
vi B = sol_2(p);
double c = clock();
a = (b - a) / CLOCKS_PER_SEC;
b = (c - b) / CLOCKS_PER_SEC;
assert(A == B);
// cout << a << "\n"; 1.563 sec
// cout << b << "\n"; 1.376 sec
}
void solve() {
int a, b;
cin >> a >> b;
cout << a + b << "\n";
}
signed main() {
test();
solve();
return 0;
}