题解 | #连通图#

连通图

https://www.nowcoder.com/practice/569e89823a5141fe8a11ab7d4da21edf

class UnionFind:#能不能改下样例自测
    def __init__(self, n):
        self.pt = [i for i in range(n)]
        self.h = [0 for i in range(n)]

    def find(self, x):
        if self.pt[x] != x:
            self.pt[x] = self.find(self.pt[x])
        return self.pt[x]

    def union(self, x, y):
        px, py = self.pt[x], self.pt[y]
        if px == py:  # 如果父节点一样
            return
        if self.h[px] > self.h[py]:
            self.pt[py] = self.find(self.pt[x])
        elif self.h[px] < self.h[py]:
            self.pt[px] = self.find(self.pt[y])
        else:
            self.pt[px] = self.find(self.pt[y])
            self.h[y] += 1


while True:
    try:
        n, m = map(int, input().split(" "))
        if m == 0:
            break
        v = UnionFind(n)
        s = "YES"
        ct = 0
        for i in range(m):
            a, b = map(int, input().split(" "))
            v.union(a - 1, b - 1)
        for i in range(n):
            if v.pt[i] == i:
                ct += 1
        if ct > 1:
            s = "NO"
        print(s)
    except:
        break

全部评论

相关推荐

不愿透露姓名的神秘牛友
11-26 15:46
已编辑
字节国际 电商后端 24k-35k
点赞 评论 收藏
分享
喜欢走神的孤勇者练习时长两年半:爱华,信华,等华,黑华
点赞 评论 收藏
分享
投递华为等公司10个岗位
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务