We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1979621 commit 5ecd764Copy full SHA for 5ecd764
library/data_structures/dsu/dsu.hpp
@@ -1,13 +1,13 @@
1
#pragma once
2
// NOLINTNEXTLINE(readability-identifier-naming)
3
struct DSU {
4
- vi e;
5
- DSU(int n): e(n, -1) {}
6
- int size(int x) { return -e[go(x)]; }
7
- int go(int x) { return e[x] < 0 ? x : e[x] = go(e[x]); }
+ vi p;
+ DSU(int n): p(n, -1) {}
+ int size(int x) { return -p[go(x)]; }
+ int go(int x) { return p[x] < 0 ? x : p[x] = go(p[x]); }
8
bool join(int a, int b) {
9
if ((a = go(a)) == (b = go(b))) return 0;
10
- if (e[a] > e[b]) swap(a, b);
11
- return e[a] += e[b], e[b] = a, 1;
+ if (p[a] > p[b]) swap(a, b);
+ return p[a] += p[b], p[b] = a, 1;
12
}
13
};
0 commit comments