#给你一个整数 n ,表示一张 无向图 中有 n 个节点,编号为 0 到 n - 1 。同时给你一个二维整数数组 #edges ,其中 edges[i] = [ai, bi] 表示节点 ai 和 bi 之间有一条 无向 边。
#请你返回 无法互相到达 的不同 点对数目 。
class Solution(object):
def countPairs(self, n, edges):
"""
:type n: int
:type edges: List[List[int]]
:rtype: int
"""
# 构建graph list, graph[i]表示i节点到哪些节点是直接连通的
graph = [[] for _ in range(n)]
for x, y in edges:
graph[x].append(y)
graph[y].append(x)
# 初始化 访问list 辅助dfs
visited = [False] * n
# 封装dfs函数,参数为x节点,从x节点出发,深搜每个与x直接连通的节点
# 节点没有被访问过,count+1, count是连通的节点个数,n-count是还没有连通的节点个数
def dfs(x):
visited[x] = True
count = 1
for y in graph[x]:
if not visited[y]:
count += dfs(y)
return count
# 由于每个连通分量和另一个连通分量每个点都无法相互到达,所以拿count * (n-count)
# 由于每个连通分量都会进行计算,所以总共计算了两遍,最后结果需要除二
res = 0
for i in range(n):
if not visited[i]:
count = dfs(i)
res += count * (n - count)
return res // 2
leetcode 2316. 统计无向图中无法互相到达点对数
于 2023-10-21 15:31:00 首次发布