[Acwing算法基础] 2.7 并查集

并查集的应用

1. 将两个集合合并
2. 询问两个元素是否在一个集合当中

并查集能以近乎O(1)的时间复杂度完成这两个操作

基本原理

使用树的形式来保存每个集合。树根的编号就是整个集合的编号，每个节点储存他的父节点，p[x]表示x的父节点

问题1：如何判断一个节点是不是树根？
答：if (p[x] == x) 那么它就是树根

问题2：如何求x的集合编号？

while (p[x] != x) x = p[x];  // 这里的时间复杂度还是很高的

优化：路径压缩
第一次找到x所属的集合（根节点）之后，直接将路径上所有点都直接指向根节点。（第二次就只需要走一步了）并查集添加这个优化之后可以看作O(1)的复杂度了。

优化：按秩合并，碰到的概率很低

问题3：如何合并两个集合？
答：若p[x]是x的集合编号，py是y的集合编号p[x] = y

核心代码

1. 集合合并操作

if (op[0] == 'M') p[find(a)] = find(b);

2. 找x的根节点

int find(int x)  // 返回x的祖宗节点 + 路径压缩
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

完整代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10;

int n, m;
int p[N];  // father数组

int find(int x)  // 返回x的祖宗节点 + 路径压缩
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    cin >> n >> m;
    
    // 1. 初始化父节点
    for (int i = 1; i <= n; ++i) p[i] = i;
    
    while (m--)
    {
        char op[2];
        int a, b;
        scanf("%s%d%d", op, &a, &b);

        if (op[0] == 'M') p[find(a)]  = find(b);  // 合并两个集合
        else
        {
            if (find(a) == find(b)) puts("Yes");
            else puts("No");
        }     
    }
    return 0;
}