There was an error while loading. Please reload this page.
1 parent 018c49e commit 1a86246Copy full SHA for 1a86246
data_structure/ufds.cpp
@@ -1,13 +1,13 @@
1
class UFDS {
2
public:
3
- vector<int> p, rank, setSizes;
+ vi p, rank, setSizes;
4
int numSets;
5
6
UFDS(int n) {
7
numSets = n;
8
rank.assign(n, 0); p.assign(n, 0);
9
- for (int i = 0; i < n; i++) p[i] = i;
10
- setSizes.assign(N, 1);
+ forn(i, n) p[i] = i;
+ setSizes.assign(n, 1);
11
}
12
int findSet(int i) { return (p[i] == i) ? i : p[i] = findSet(p[i]); }
13
bool isSameSet(int i, int j) { return findSet(i) == findSet(j); }
@@ -25,5 +25,4 @@ class UFDS {
25
26
27
int setSize(int i) { return setSizes[findSet(i)]; }
28
- void clear(){ p.clear(); rank.clear(); setSizes.clear(); }
29
};
0 commit comments