[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;
    cin >> n >> m;
    for (int i = 1; i <= n; i ++)   p[i] = i;
    while (m --) {
        char op;
        int a, b;
        cin >> op >> a >> b;
        if (op == 'M')  p[find(a)] = find(b);
        else {
            if (find(a) == find(b))     puts("Yes");
            else    puts("No");
        }
    }
    return 0;
}

---

1. p[ x ] 用来存 x 的父节点编号，根节点满足 p[ x ] = x；
2. find 作用是返回根节点的值，并把路径上所有点的父节点都变为根节点（路径压缩）；
3. 合并集合 a 和 b，将 a 集合根节点变为 b 集合根节点的子节点；