LeetCode:261. 以图判树 - Python

261. 以图判树

问题描述:

给定从 0 n-1 标号的 n 个结点,和一个无向边列表(每条边以结点对来表示),请编写一个函数用来判断这些边是否能够形成一个合法有效的树结构。

示例 1:

输入:n = 5, 边列表 edges = [[0,1], [0,2], [0,3], [1,4]]
输出:true

示例 2:

输入:n = 5, 边列表 edges = [[0,1], [1,2], [2,3], [1,3], [1,4]]
输出:false

注意:
你可以假定边列表 edges 中不会出现重复的边。由于所有的边是无向边,边 [0,1] 和边 [1,0] 是相同的,因此不会同时出现在边列表 edges 中。

问题分析:

这题目有点贵呀,是LeetCode的VIP题目,第一次见还有点蒙,其实仔细想想也没啥难的。问题分析,判断一个无向图能否勾成一个树,很显然这个图要满足3个条件:

  1. 这个图不存在环
  2. 这个图所有节点是连通
  3. 这个图的边数一定为 n-1, 因为如果一棵树有n个节点,那么它的边一定是n-1
  4. 是不是可以得出这样的结论:如果有n-1条边且有环是一定是不连通,是不是可以说明,在n-1条边的条件下,只要判断是否有环即可?没有环路边数为n-1,就一定能构造成树?(没有严谨的证明哈,感觉反证法可以证明)

现在看看题目如何做?
(1)第一个条件就是判断这个图的边数是否等于n-1,很显然不符合就直接返回 False 即可。
(2)使用并查集的思想判断是否存在环路,如果存在环路直接返回 False,否则最后就返回 True

Python3实现:

# @Time   :2023/09/06
# @Author :Liu


class Solution:

    def validTree(self, n, edges):

        if len(edges) != n - 1:  # 边数是否等于 n - 1
            return False

        def find(x):  # 并查集查找
            if fa[x] != x:
                fa[x] = find(fa[x])
            return fa[x]

        fa = [i for i in range(n)]
        for x, y in edges:  # 判断两个点是否在同一个并查集里面
            fa_x = find(x)
            fa_y = find(y)

            if fa_x == fa_y:
                return False

            fa[fa_x] = fa_y

        return True


if __name__ == '__main__':
    solu = Solution()
    n, edges = 7, [[0, 1], [1, 2], [2, 3], [4, 5], [4, 6], [5, 6]]
    print(solu.validTree(n, edges))

相关参考:
[1]LeetCode:261. 以图判树VIP 题目,反正我是打不开。
[2] 代码参考: yiduobo的每日leetcode 261.以图判树。只在本地验证了,没有在线验证。
声明: 总结学习,有问题或不当之处,可以批评指正哦,谢谢。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值