数据结构用结构体实现并查集

Posted 2020-11-27 jiangyuechen

struct union_find_set
{
    int father[MAXN];//每个元素的祖先
    union_find_set(int limit)//构造函数，对于每一个元素，其父亲初始化为它本身
    {
        for(int i=1;i<=limit;i++)
        {
            father[i]=i;
        }
    }
    int find(int x)//找寻操作，使用路径压缩优化
    {
        if(father[x]!=x) father[x]=find(father[x]);
        return father[x];
    }
    void unite(int a,int b)//合并操作，实质是将一个集合的共同父亲赋值为另一个集合
    {
        father[b]=a;
    }
    bool isSameSet(int a,int b)//判断是否在同一个集合里
    {
        return (find(a)==find(b));//只要两个元素的祖先相同就可以
    }
};