【算法基础】并查集 AcWing 836. 合并集合

---

一、题目&解读

1、题目

题目链接{点击跳转}

二、思路

1、编号

从1开始

for (int i = 1; i <= n;i ++)
        p[i] = i;//编号

2、加入集合

合并操作就是将a的父亲结点 变为b的父亲结点

p[find(a)] = find(b)//合并

这个语句可以这样子理解 原先p[a]=a p[b]=b
合并操作后p[a]=p[b]=b
再合并c也是一样的p[a]=p[b]=p[c]=b
数学可以这样表示b={a,b,c}
b称为祖宗结点 所以上面就是一个祖宗结点为b的集合

3、合并俩个集合

经过上面的理解后，合并集合也就很easy了
比如现在 a集合 ：a={1,2,a} b集合：b={3,4,b}
直接p[find(a)] = find(b) 就行了 因为b里面所以数只是编号不同但是祖宗结点的值是一样的,如下图

4、判断是否在同一个集合

find(a) == find(b)只需比较祖宗结点的值是否相同即可

三、AC代码

时间复杂度 近乎 O(1)

#include <iostream>
using namespace std;
const int N = 1000010;
int n, m;
int p[N];//父亲结点

//核心算法：
int find (int x) {
    if (p[x] != x)
        p[x] = find(p[x]);//路径压缩
    return p[x];//返回祖宗结点
}

int main () {
    scanf("%d%d", &n, &m);

    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);//把a加到b的集合中
        else {
            if (find(a) == find(b))//判断a和b的父亲结点是否相同
                puts("Yes");
            else
                puts("No");
        }
    }
    return 0;
}

四、总结

学数据结构 最主要的还是要画图，先画一遍，代码自然就能够写了。

结尾：

感谢你能看完，希望对你有帮助 ，如有错误欢迎指正，码字不易，给个赞呗