寻找图中是否存在路径

1971. 寻找图中是否存在路径 - 力扣（LeetCode）

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

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

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

示例 1：

输入：n = 3, edges = [[0,1],[1,2],[2,0]], source = 0, destination = 2
输出：true
解释：存在由顶点 0 到顶点 2 的路径:
- 0 → 1 → 2 
- 0 → 2

示例 2：

输入：n = 6, edges = [[0,1],[0,2],[3,5],[5,4],[4,3]], source = 0, destination = 5
输出：false
解释：不存在由顶点 0 到顶点 5 的路径.

提示：

• 1 <= n <= 2 * 105
• 0 <= edges.length <= 2 * 105
• edges[i].length == 2
• 0 <= ui, vi <= n - 1
• ui != vi
• 0 <= source, destination <= n - 1
• 不存在重复边
• 不存在指向顶点自身的边

使用并查集可以判断两个顶点是否连通

class Solution {
    public boolean validPath(int n, int[][] edges, int source, int destination) {
        UnionFind uf = new UnionFind(n);
        for(int[] edge :edges) {
            uf.join(edge[0], edge[1]);
        }
        return uf.isSame(source, destination);
    }

    public static class UnionFind{
        private int[] father;
        public UnionFind(int n) {
            father = new int[n];
            for(int i = 0; i < n; i++){
                father[i] = i;
            } 
        }

        public int find(int u) {
            if(father[u] == u) return u;
            father[u] = find(father[u]); //路径优化，让u的父节点指向根
            return father[u];
        }

        public boolean isSame(int u, int v) {
            return find(u) == find(v);
        }

        public void join(int u, int v) {
            u = find(u); //找u的父亲
            v = find(v); //找v的父亲
            if(u != v) {
                father[v] = u;
            }
        }
    }
}