/tmp/solutions/build/unionfind-main.cpp:
1| |#include <common.h>
2| |prelude;
3| |
4| |namespace {
5| |
6| |int pa[200000];
7| |
8| 51.6k|def find(int u) -> int {
9| 51.6k| int w = pa[u], v;
10| 51.6k| if (w < 0) return u;
^49.6k
------------------
| Branch (10:7): [True: 95.99%, False: 4.01%]
------------------
11| 2.07k| while (pa[w] >= 0) w = pa[w];
^2.07k ^6
------------------
| Branch (11:10): [True: 0.29%, False: 99.71%]
------------------
12| 2.07k| while (pa[u] != w) v = u, u = pa[u], pa[v] = w;
^6
------------------
| Branch (12:10): [True: 0.29%, False: 99.71%]
------------------
13| 2.07k| return w;
14| 51.6k|};
15| |
16| |} // namespace
17| |
18| 1|int main() {
19| 1| rd rd;
20| 1| wt wt;
21| 1| int n = rd.uh();
22| 1| int q = rd.uh();
23| 1| std::memset(pa, -1, 4 * n);
24| 25.8k| while (q--) {
------------------
| Branch (24:10): [True: 100.00%, False: 0.00%]
------------------
25| 25.8k| let t = rd.u1();
26| 25.8k| let u = find(rd.uh());
27| 25.8k| let v = find(rd.uh());
28| 25.8k| if (t == 0 && u != v) {
^13.0k
------------------
| Branch (28:9): [True: 50.40%, False: 49.60%]
| Branch (28:19): [True: 100.00%, False: 0.00%]
------------------
29| 13.0k| if (pa[u] < pa[v])
------------------
| Branch (29:11): [True: 6.81%, False: 93.19%]
------------------
30| 887| pa[v] = u;
31| 12.1k| else if (pa[v] < pa[u])
------------------
| Branch (31:16): [True: 7.34%, False: 92.66%]
------------------
32| 891| pa[u] = v;
33| 11.2k| else
34| 11.2k| pa[u] = v, --pa[v];
35| 13.0k| }
36| 25.8k| if (t == 1) {
------------------
| Branch (36:9): [True: 49.60%, False: 50.40%]
------------------
37| 12.8k| wt.u1(u == v);
38| 12.8k| }
39| 25.8k| }
40| 1| return 0;
41| 1|}