LeetCode 1971. 寻找图中是否存在路径

原创

Daylight629 2023-01-31 14:31:37 博主文章分类：LeetCode题解 ©著作权

©著作权归作者所有：来自51CTO博客作者Daylight629的原创作品，请联系作者获取转载授权，否则将追究法律责任

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

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

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

示例 1：

LeetCode 1971. 寻找图中是否存在路径_数组

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

示例 2：

LeetCode 1971. 寻找图中是否存在路径_leetcode_02

输入：n = 6, edges = [[0,1],[0,2],[3,5],[5,4],[4,3]], start = 0, end = 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 <= start, end <= n - 1
不存在双向边
不存在指向顶点自身的边

二、方法一

并查集

class Solution {
    public boolean validPath(int n, int[][] edges, int source, int destination) {
         Check check = new Check(n);
         for (int i = 0; i < edges.length; i++) {
            check.merge(edges[i][0], edges[i][1]);
         }
         return check.isConnect(source, destination);
    }
    class Check{
        int[] p;
        int size;
        public Check(int size) {
            this.size = size;
            p = new int[size];
            for (int i = 0; i < size; i++) {
                p[i] = i;
            }
        }
        public int find(int a) {
            int b = p[a];
            if (a != b) return find(b);
            return b;
        }
        public boolean isConnect(int a, int b) {
            return find(a) == find(b);
        }
        public void merge(int a, int b) {
            int c = find(a);
            int d = find(b);
            if (d == c) return;
            p[c] = d;
        }
    }
}