LeetCode 1971 寻找图中是否存在路径——并查集初学

题目:

有一个具有 n个顶点的 双向 图，其中每个顶点标记从 0 到 n - 1（包含 0 和 n - 1）。图中的边用一个二维整数数组 edges 表示，其中 edges[i] = [ui, vi] 表示顶点 ui 和顶点 vi 之间的双向边。 每个顶点对由 最多一条 边连接，并且没有顶点存在与自身相连的边。

请你确定是否存在从顶点 start 开始，到顶点 end 结束的 有效路径 。

给你数组 edges 和整数 n、start和end，如果从 start 到 end 存在 有效路径 ，则返回 true，否则返回 false 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-if-path-exists-in-graph

 

思路一：

先用一个n*n的数组创建出图，再使用bfs从起点出发遍历整个图，直到遍历到终点为止，返回true,若遍历结束也未达到终点,则返回false。

代码:

class Solution:
    from collections import deque
    def validPath(self, n: int, edges: List[List[int]], source: int, destination: int) -> bool:
        if n == 1: return True
        q = deque([source])
        visited = [0 for i in range(n)]
        graph = [[0 for i in range(n)] for j in range(n)]
        res = False
        for i in edges:
            graph[i[0]][i[1]] = 1
            graph[i[1]][i[0]] = 1
        while q:
            p = q.popleft()
            for i in range(n):
                if graph[p][i] and not visited[i]:
                    visited[i] = 1
                    if i == destination: return True
                    q.append(i)
        return res
            

        

结果:在30000个点的用例时超时

显然直接上bfs是不能的,在我思考是否使用dfs再做一次时，翻题解发现有使用并查集的做法，之前并未听说过并查集,便开始学习并查集,并查集具体内容我就不说了，因为自己也是参考别人的，这里给出两篇参考文章：

菜鸟教程: 并查集快速合并 | 菜鸟教程

知乎: 算法学习笔记(1) : 并查集 - 知乎

思路二：

使用并查集处理所有给出的边，最后通过并查集查询根判断起点个终点是否属于一个集合，若根相同则返回true，否则返回false。

代码:

class Solution:
    from collections import deque
    def validPath(self, n: int, edges: List[List[int]], source: int, destination: int) -> bool:
        num = [i for i in range(n)]

        def get(node):
            if num[node] == node:
                return node 
            num[node] = get(num[node]) #次要改动
            return num[node]

        def merge(m, n):
            m_get, n_get = get(m), get(n)
            if m_get != n_get: 
                num[n_get] = m_get #主要改动

        for edge in edges:
            merge(edge[0], edge[1])
        
        return get(source) == get(destination)

结果:

这里附上一个测试用例的num数组的具体过程，以便日后理解:

n = 10

edges = [[0,7],[0,8],[6,1],[2,0],[0,4],[5,8],[4,7],[1,3],[3,5],[6,5]]