使用等级和路径压缩试探法合并数据库表

Question

我在数据结构课程中遇到以下问题。 该课程提供的错误相当模糊，我无法辨别错误所在。

注意：错误消息仅显示“错误答案”。 并且没有提供测试用例。

输入格式：输入的第一行包含两个整数和——分别是数据库中表的数量和要执行的合并查询的数量。 输入的第二行包含整数 [] — 第 - 个表中的行数。 然后下面几行描述了合并查询。 它们每个都包含两个整数 n[i] 和 [i]——要合并的表的数量。

Output 格式：对于每个查询，打印一行包含单个 integer — 相应操作后所有表的最大大小（根据行数）。

示例输入：

5 5
1 1 1 1 1
3 5
2 4
1 4
5 4
5 3

样本 Output：

2
2
3
5
5

这是我当前的代码，它适用于大多数情况，但似乎有一些我没有考虑到的边缘情况。

class DataBases:
    def __init__(self, row_counts):
        self.max_row_count = max(row_counts)
        self.row_counts = row_counts
        n_tables = len(row_counts)
        self.parent = list(range(n_tables))
        self.rank = [1] * n_tables

    def get_parent(self, table):
        update_root = []
        root = table
        
        while root != self.parent[root]:
            update_root.append(self.parent[root])
            root = self.parent[root]
            
            for i in update_root:
                self.parent[i] = root
                
        return root

    def merge_tables(self, dst, src):
        src_parent = self.get_parent(src)
        dst_parent = self.get_parent(dst)
        if src_parent == dst_parent: return
        
        if self.rank[src_parent] > self.rank[dst_parent]:
            self.parent[dst_parent] = src_parent
            self.update_row_counts(src_parent, dst_parent)
        else:
            self.parent[src_parent] = dst_parent
            self.update_row_counts(dst_parent, src_parent)
            if self.rank[src_parent] == self.rank[dst_parent]:
                self.rank[dst_parent] += 1
    
    def update_row_counts(self, root, child):
        self.row_counts[root] += self.row_counts[child]
        self.row_counts[child] = 0
        self.max_row_count = max(self.max_row_count, self.row_counts[root])

def main():
    n_tables, n_queries = map(int, input().split())
    counts = list(map(int, input().split()))
    assert(n_tables == len(counts))
    db = DataBases(counts)
    for i in range(n_queries):
        dst, src = map(int, input().split())
        db.merge_tables(dst - 1, src - 1)
        print(db.max_row_count)



if __name__ == "__main__":
    main()

Answer 1

问题出在 get_parent（路径压缩）实现中。

正确的解决方案：

    def get_parent(self, table):
        
        if table != self.parent[table]:
            self.parent[table] = self.get_parent(self.parent[table])
                
        return self.parent[table]