547. 省份数量

题目描述:有 n 个城市,其中一些彼此相连,另一些没有相连。如果城市 a 与城市 b 直接相连,且城市 b 与城市 c 直接相连,那么城市 a 与城市 c 间接相连。
省份 是一组直接或间接相连的城市,组内不含其他没有相连的城市。
给你一个 n x n 的矩阵 isConnected ,其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连,而 isConnected[i][j] = 0 表示二者不直接相连。
返回矩阵中 省份 的数量。
解题思路一:DFS

class Solution:
    def findCircleNum(self, M: List[List[int]]) -> int:
        n = len(M)
        if n == 0: return 0
        res = 0
        def dfs(i):
            M[i][i] = 0
            for j in range(n):
                if M[i][j] and M[j][j]:
                    dfs(j)
        for i in range(n):
            if M[i][i]:
                res += 1
                dfs(i)
        return res

解题思路二:并查集

class Solution:
    def findCircleNum(self, M: List[List[int]]) -> int:
        n = len(M)
        s = list(range(n))
        def find(i):
            root = s[i]
            while root != s[root]:
                root = s[root]
            while root != i:
                originroot = s[i]
                s[i] = root
                i = originroot
            return s[i]
        for i in range(n):
            for j in range(i+1, n):
                if M[i][j]:
                    root_i = find(i)
                    root_j = find(j)
                    if root_i != root_j:
                        s[root_j] = root_i

            
        for i in range(n):
            find(i)
        return len(set(s))
                    
                    
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值