【C++】Day21 并查集 AcWing 836. 合并集合 (算法基础课笔记)


前言
你好啊,我最近在学acwing的算法基础课,备战蓝桥杯,如果你也是一样的话,欢迎一起学习~


一、题目&解读

1、题目

题目链接{点击跳转}

2、题目解读

在这里插入图片描述

二、思路

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)

扫描二维码关注公众号,回复: 13712134 查看本文章
#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;
}

四、总结

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

结尾:

感谢你能看完,希望对你有帮助 ,如有错误欢迎指正,码字不易,给个赞呗 在这里插入图片描述

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_49486457/article/details/123009849