题目描述:有 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))