并查集-路径压缩法

对并查集的理解

我认为的并查集是：每一个集合抽象成一颗N叉数， 树上的每个节点存储的是双亲结点的位置，只有根节点存储的是自身编号，根节点代表了这棵树(即这个集合)

并查集可以用来维护无向图的连通块,也可以用来解决具有传递性质的问题

并查集主要有两种操作
合并：将集合A根节点的值赋给集合B根节点，这样集合B的所有节点往上遍历时都会找到 集合A的根节点
查询：找出这两颗树的根节点，判断是否相等，如果相等则是同一个集合，否则不是同一个集合

图解

从上图可见,把2所在的集合合并到6所在的集合后,2所在的集合的根节点(这里仍然是2)里面的值就会变成6,图的右侧也发生了对应的变化

从上图可见,把3所在的集合合并到6所在的集合后,3所在的集合的根节点(这里仍然是3)里面的值就会变成6,图的右侧也发生了对应的变化

从上图可见,把6所在的集合合并到5所在的集合后,6所在的集合的根节点(这里仍然是6)里面的值就会变成5,图的右侧也发生了对应的变化;这时要特别注意,2和3还没有改变,仍然指向6

从上图可见,只有到使用3的时候才会把3里面的值变成5,这时3就指向5了

并查集模板题

AcWing 836. 合并集合
代码：

#include<iostream>

using namespace std;
const int N = 1e5 + 10;
int p[N];

//并查集函数, 使用路径压缩法
int find(int x) {
    if (p[x] != x)
        p[x] = find(p[x]);
    return p[x];
}

int main() {
    int n, m;
    scanf("%d%d", &n, &m);  //输入集合个数和操作次数
    
    //初始化并查集
    for (int i = 1; i <= n; i++) p[i] = i;
    
    while (m--) {
        string op;
        cin >> op;
        int a, b;
        cin >> a >> b;
        
        //如果输入是M,则进行合并操作
        if (op == "M") p[find(a)] = find(b);
        //如果输入是Q,则进行查询操作
        else {
            if (find(a) == find(b)) puts("Yes");
            else puts("No");
        }
    }
    return 0;
}