diff --git a/data_structures/disjoint_set/disjoint_set.py b/data_structures/disjoint_set/disjoint_set.py index a93b89621c4a..f3c5e4f9b83e 100644 --- a/data_structures/disjoint_set/disjoint_set.py +++ b/data_structures/disjoint_set/disjoint_set.py @@ -28,10 +28,10 @@ def union_set(x, y): x, y = find_set(x), find_set(y) if x.rank > y.rank: y.parent = x + elif x.rank == y.rank: + y.rank += 1 else: x.parent = y - if x.rank == y.rank: - y.rank += 1 def find_set(x):