并查集模板及思想

最新推荐文章于 2024-10-01 04:50:10 发布

葫芦大仙

最新推荐文章于 2024-10-01 04:50:10 发布

阅读量104

点赞数

文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/qq_46689648/article/details/126130745

版权

本文介绍了并查集数据结构的基本思想和优化方法——路径压缩，通过ACwing上的题目展示了如何使用C++实现并查集，并在合并集合和查找元素所属集合时保持高效。此外，还探讨了如何在并查集中拓展维护集合的额外信息，如计数，以解决实际问题。

摘要由CSDN通过智能技术生成

基本思想

如果想合并两个集合，直接把根节点插到另一个集合的某个地方即可

路径压缩优化–时间复杂度近乎为O(1)

在问题2中，求x的集合编号与数的高度成正比，复杂度较高，所以可以进行优化，在第一次查的时候正常一步一步查到根节点while(p[x]!=x) x=p[x]，但一旦找到后就会把这个线上的所有点都指向根节点，之后再查就是O(1)的复杂度

代码

模板题：https://www.acwing.com/problem/content/838/

#include <iostream>
using namespace std;
const int N = 100010;
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--)
    {
        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)) cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
    }
    return 0;
}

拓展维护集合额外信息

维护集合个数

:::info
例题：https://www.acwing.com/activity/content/problem/content/886/
:::

#include <iostream>
using namespace std;
const int N = 100010;
int p[N],cnt[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;
        cnt[i] = 1;
    }
    
    while(m--) 
    {
        char op[5];
        int a,b;
        scanf("%s",op);
        if(op[0]=='C') 
        {
            scanf("%d%d",&a,&b);
            a = find(a);
            b=find(b);
            if (a!=b){
                p[a] = b;
                cnt[b]+=cnt[a];
            } 
            
        }else{
            if (op[1] == '1') 
            {
                scanf("%d%d",&a,&b);
                if(find(a)==find(b))
                    cout<<"Yes"<<endl; 
                else 
                    cout<<"No"<<endl; 
            }else{
                scanf("%d",&a);
                cout<<cnt[find(a)]<<endl;
            }
                
        }
    }
    return 0;
}