并查集与路径压缩

基本概念

并查集是一种树型的数据结构，用于处理一些不相交集合的合并及查询问题。

并查集的思想是用一个数组表示了整片森林（parent），树的根节点唯一标识了一个集合，我们只要找到了某个元素的的树根，就能确定它在哪个集合里。

假设要把N个元素归为若干的集合，并实现以下功能：

union(p,q) - 将 p 和 q 两个元素连接起来。

find(p) - 查询 p 元素在哪个集合中。

isConnected(p,q) - 查看 p 和 q 两个元素是否相连接在一起。

把每一个元素，看做是一个节点并且指向自己的父节点，根节点指向自己。

判断两个元素是否连接，只需判断根节点是否相同。

连接两个元素，只需找到他们对应的根节点，让根节点相连即可。

构建这种指向父节点的树形结构， 使用一个数组构建一棵指向父节点的树，parent[i] 表示 i 元素所指向的父节点。

路径压缩

并查集里的 find 函数里可以进行路径压缩，是为了更快速的查找一个点的根节点。对于一个集合树来说，它的根节点下面可以依附着许多的节点，因此，我们可以尝试在 find 的过程中，从底向上，如果此时访问的节点不是根节点的话，那么我们可以把这个节点尽量的往上挪一挪，减少数的层数，这个过程就叫做路径压缩。

参考文献：并查集路径压缩 | 菜鸟教程 (runoob.com)https://www.runoob.com/data-structures/union-find-compress.html

将路径压缩至一层的方法:

...
// 查找过程, 查找元素p所对应的集合编号
// O(h)复杂度, h为树的高度
int find(int p) {

    //深度最浅路径压缩算法
    if (p != parent[p])
        parent[p] = find(parent[p]);
    return parent[p];
}
...

参考例题

题目详情 - 05-树8 File Transfer (25 分) (pintia.cn)https://pintia.cn/problem-sets/1497448825169559552/problems/1508006239265828865题目大意：

第一行输入元素的个数

之后的各行，“I c1 c2 ”表示将C1 C2两个节点连接起来，“C c1 c2”表示监测两节点是否可以联通。

#include <iostream>
using namespace std;
const int maxn = 10005;
int fa[maxn];
int N, x, y;
char ch;
inline void init() {
    for(int i = 1; i <= N; ++i) 
        fa[i] = i;
}
int find(int x) {//查询+路径压缩 把沿途的每个节点的父节点都设为根节点
    if(x!=fa[x]){
        fa[x]=find(fa[x]);
    }
    return fa[x];
}
void Merge(int x, int y) {
    fa[find(x)] = find(y);
}
int main() {
    scanf("%d", &N);
    init();
    getchar();
    scanf("%c", &ch);
    while(ch != 'S') {
        scanf("%d %d", &x, &y);
        getchar();
        if(ch == 'C') {
            if(find(x) == find(y)) cout << "yes" << endl;
            else cout << "no" << endl;
        } else if(ch == 'I') Merge(x, y);
        scanf("%c", &ch);
    }
    int ans = 0;
    for(int i = 1; i <= N; ++i) {
        if(fa[i] == i) ans++;
    }
    if(ans == 1) cout << "The network is connected." << endl;
    else cout << "There are " << ans << " components." << endl;
    return 0;
}