图论篇--代码随想录算法训练营第五十五天打卡| 并查集理论，107. 寻找存在的路径

并查集理论

并查集主要有三个功能

1. 寻找根节点，函数：find(int u)，也就是判断这个节点的祖先节点是哪个
2. 将两个节点接入到同一个集合，函数：join(int u, int v)，将两个节点连在同一个根节点上
3. 判断两个节点是否在同一个集合，函数：isSame(int u, int v)，就是判断两个节点是不是同一个根节点

模板 

int n = 1005; // n根据题目中节点数量而定，一般比节点数量大一点就好
vector<int> father = vector<int> (n, 0); // C++里的一种数组结构

// 并查集初始化
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]); // 路径压缩，让除了根节点其他所有节点都挂载根节点下
}

// 判断 u 和 v是否找到同一个根
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); // 寻找u的根
    v = find(v); // 寻找v的根
    if (u == v) return ; // 如果发现根相同，则说明在一个集合，不用两个节点相连直接返回
    father[v] = u;
}

107. 寻找存在的路径

题目链接：107. 寻找存在的路径

题目描述：

给定一个包含 n 个节点的无向图中，节点编号从 1 到 n （含 1 和 n ）。你的任务是判断是否有一条从节点 source 出发到节点 destination 的路径存在。

解题思路：

并查集直接应用

代码：

#include<iostream>
#include<vector>
using namespace std;

vector<int> father;
void init()
{
    for(int i = 0 ; i < father.size(); i++)
        father[i] = i;
}

int find(int s)
{
    return s == father[s] ? s : father[s] = find(father[s]);
}

void join(int s, int t)
{
    s = find(s);
    t = find(t);
    if(s == t) return;
    father[s] = t;
}

int main()
{
    int n,m;
    cin >> n >> m;
    father.resize(n+1);
    init();
    for(int i = 0 ; i < m; i++)
    {
        int s, t;
        cin >> s >> t;
        join(s,t);
    }
    int source, destination;
    cin >> source >> destination;
    if(find(source) == find(destination)) cout << 1 << endl;
    else cout << 0 << endl;
    return 0;
}