【代码随想录算法训练营第42期第五十五天 | 寻找存在的路径】-CSDN博客

本文链接：https://blog.csdn.net/cattte/article/details/140224736

代码随想录算法训练营第42期第五十五天 | 寻找存在的路径

一、寻找存在的路径

解题代码C++：

#include <iostream>
#include <vector>

using namespace std;

int n; 
vector<int> father = vector<int> (101, 0); 

// 并查集初始化
void init() {
    for (int i = 1; i <= n; i++)  father[i] = i;
}
// 并查集里寻根的过程
int find(int u) {
    return u == father[u] ? u : father[u] = find(father[u]);
}

// 判断 u 和 v是否找到同一个根
bool isSame(int u, int v) {
    u = find(u);
    v = find(v);
    return u == v;
}

void join(int u, int v) {
    u = find(u);
    v = find(v);
    if (u == v) return ; // 如果发现根相同，则说明在一个集合，不用两个节点相连直接返回
    father[v] = u;
}

int main() {
    int m, s, t, source, destination;
    cin >> n >> m;
    init();
    while (m--) {
        cin >> s >> t;
        join(s, t);
    }
    cin >> source >> destination;
    if (isSame(source, destination)) cout << 1 << endl;
    else cout << 0 << endl;
}