Skip to content

Commit 1a86246

Browse files
author
alirezakhm
authored
Update ufds.cpp
1 parent 018c49e commit 1a86246

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

data_structure/ufds.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
class UFDS {
22
public:
3-
vector<int> p, rank, setSizes;
3+
vi p, rank, setSizes;
44
int numSets;
55

66
UFDS(int n) {
77
numSets = n;
88
rank.assign(n, 0); p.assign(n, 0);
9-
for (int i = 0; i < n; i++) p[i] = i;
10-
setSizes.assign(N, 1);
9+
forn(i, n) p[i] = i;
10+
setSizes.assign(n, 1);
1111
}
1212
int findSet(int i) { return (p[i] == i) ? i : p[i] = findSet(p[i]); }
1313
bool isSameSet(int i, int j) { return findSet(i) == findSet(j); }
@@ -25,5 +25,4 @@ class UFDS {
2525
}
2626
}
2727
int setSize(int i) { return setSizes[findSet(i)]; }
28-
void clear(){ p.clear(); rank.clear(); setSizes.clear(); }
2928
};

0 commit comments

Comments
 (0)