LeetCode1971. 寻找图中是否存在路径

LeetCode1971. 寻找图中是否存在路径

---

题目：

• 有一个具有 n 个顶点的 双向 图，其中每个顶点标记从 0 到 n - 1（包含 0 和 n - 1）。图中的边用一个二维整数数组 edges 表示，其中 edges[i] = [$u_i$, $v_i$] 表示顶点 $u_i$ 和顶点 $v_i$ 之间的双向边。 每个顶点对由 最多一条 边连接，并且没有顶点存在与自身相连的边。
• 请你确定是否存在从顶点 source 开始，到顶点 destination 结束的 有效路径 。
• 给你数组 edges 和整数 n、source 和 destination，如果从 source 到 destination 存在 有效路径 ，则返回 true，否则返回 false 。

示例：

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

解题思路：并查集
 用join()方法将所有边加入并查集，用isSame()方法判断source与destination是否属于同一集合，若是则代表有路径，返回true，否则返回false。

C++版整体代码（并查集）：

class Solution {
public:
    int n = 200005;
    vector<int> father = vector<int>(n, 0);
    // 并查集初始化
    void init(){
        for(int i = 0; i < n; i++) father[i] = i;
    }
    // 并查集寻根操作
    int find(int u){
        return u == father[u] ? u : father[u] = find(father[u]);
    }
    // 判断两个节点是否属于同一集合
    bool isSame(int u, int v){
        u = find(u);
        v = find(v);
        return u == v;
    }
    // 将v->u这条边加入并查集
    void join(int u, int v){
        u = find(u);
        v = find(v);
        if(u == v) return;
        father[v] = u;
    }
    bool validPath(int n, vector<vector<int>>& edges, int source, int destination) {
        init();
        for(int i = 0; i < edges.size(); i++){
            join(edges[i][0], edges[i][1]);
        }
        return isSame(source, destination);
    }
};