更新并查集相关内容

itcharge · Jun 27, 2024 · bdf258d · bdf258d
1 parent 00696fe
commit bdf258d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Contents/07.Tree/05.Union-Find/01.Union-Find.md b/Contents/07.Tree/05.Union-Find/01.Union-Find.md
@@ -251,7 +251,7 @@ class UnionFind:
             self.fa[root_y] = root_x                # y 的根节点连接到 x 的根节点上，成为 x 的根节点的子节点
         else:                                       # x 的根节点对应的树的深度 等于 y 的根节点对应的树的深度
             self.fa[root_x] = root_y                # 向任意一方合并即可
-            rank[root_y] += 1                       # 因为层数相同，被合并的树必然层数会 +1
+            self.rank[root_y] += 1                  # 因为层数相同，被合并的树必然层数会 +1
         return True
 
     def is_connected(self, x, y):                   # 查询操作：判断 x 和 y 是否同属于一个集合
@@ -389,7 +389,7 @@ class UnionFind:
             self.fa[root_y] = root_x                # y 的根节点连接到 x 的根节点上，成为 x 的根节点的子节点
         else:                                       # x 的根节点对应的树的深度 等于 y 的根节点对应的树的深度
             self.fa[root_x] = root_y                # 向任意一方合并即可
-            self.rank[y] += 1                       # 因为层数相同，被合并的树必然层数会 +1
+            self.rank[root_y] += 1                  # 因为层数相同，被合并的树必然层数会 +1
         return True
 
     def is_connected(self, x, y):                   # 查询操作：判断 x 和 y 是否同属于一个集合

diff --git a/Templates/07.Tree/Tree-UnionFind-UnoinByRank.py b/Templates/07.Tree/Tree-UnionFind-UnoinByRank.py
@@ -21,7 +21,7 @@ def union(self, x, y):                          # 合并操作：令其中一个
             self.fa[root_y] = root_x                # y 的根节点连接到 x 的根节点上，成为 x 的根节点的子节点
         else:                                       # x 的根节点对应的树的深度 等于 y 的根节点对应的树的深度
             self.fa[root_x] = root_y                # 向任意一方合并即可
-            rank[y] += 1                            # 因为层数相同，被合并的树必然层数会 +1
+            self.rank[root_y] += 1                  # 因为层数相同，被合并的树必然层数会 +1
         return True
 
     def is_connected(self, x, y):                   # 查询操作：判断 x 和 y 是否同属于一个集合