leetcode 2316. 统计无向图中无法互相到达点对数

#给你一个整数 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