subcrip
/
cp-templates
1
0
Fork 0
Code

Update trees/quick_union.cc 

Signed-off-by: subcrip <contact@subc.rip>
This commit is contained in:
subcrip 2024-04-24 11:23:13 +08:00
parent 1629a9f47f
commit 9a8c1277fb
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
trees/quick_union.cc
View File

@ -1,9 +1,10 @@
class quick_union {
private:
vector<size_t> c;
vector<size_t> c, sz;
public:
quick_union(size_t n) : c(n) {
quick_union(size_t n) : c(n), sz(n) {
iota(c.begin(), c.end(), 0);
sz.assign(n, 1);
}
size_t query(size_t i) {
@ -12,11 +13,15 @@ public:
}
void merge(size_t i, size_t j) {
sz[query(j)] += sz[query(i)];
c[query(i)] = query(j);
}
bool connected(size_t i, size_t j) {
return query(i) == query(j);
}
};
size_t query_size(size_t i) {
return sz[query(i)];
}
};