ambroisie/blog
ambroisie/blog
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

posts: union-find: fix union of same root
All checks were successful
ci/woodpecker/push/deploy/2 Pipeline was successful
Details
ci/woodpecker/cron/deploy/2 Pipeline was successful
Details

Browse source
This commit is contained in:
Bruno BELANYI 2024-06-27 13:16:44 +00:00
parent adc514215c
commit 7873828c4c
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
content/posts/2024-06-24-union-find/index.md
View file

@ -142,6 +142,9 @@ _approximate_ height, we can keep the trees balanced when merging them.
def union(self, lhs: int, rhs: int) -> int: def union(self, lhs: int, rhs: int) -> int:
lhs = self.find(lhs) lhs = self.find(lhs)
rhs = self.find(rhs) rhs = self.find(rhs)
# Bail out early if they already belong to the same set
if lhs == rhs:
return lhs
# Always keep `lhs` as the taller tree # Always keep `lhs` as the taller tree
if (self._rank[lhs] < self._rank[rhs]) if (self._rank[lhs] < self._rank[rhs])
lhs, rhs = rhs, lhs lhs, rhs = rhs, lhs