并查集

最新推荐文章于 2024-07-29 20:58:50 发布

墨白纸黑

最新推荐文章于 2024-07-29 20:58:50 发布

阅读量101

点赞数 1

分类专栏：数据结构与算法算法

本文链接：https://blog.csdn.net/qq_43487996/article/details/97886898

版权

数据结构与算法算法专栏收录该内容

28 篇文章 0 订阅

订阅专栏

/*当前集合的根就是这个集合的祖先*/
#include <iostream>
using namespace std;

int father[10000+10];//下标代表儿子，数值代表当前儿子的父亲

void Unoin(int x,int y);//合并集合
int Find(int x);//查找祖先

int main()
{
    int n,m,q;
    cin>>n>>m>>q;
    int a,b;
    int c,d;
    for(int i=1;i<=n;i++)
    {
        father[i]=i;//祖先的父亲是自己
    }
    for(int i=1;i<=m;i++)
    {
        cin>>a>>b;
        if(Find(a)!=Find(b))
            Union(a,b);
    }
    for(int i=1;i<=q;i++)
    {
        cin>>c>>d;
        if(Find(c)==Find(d))
            cout<<"yes"<<endl;
        else cout<<"no"<<endl;
    }
    return 0;
}

int Find(int x)
{
    int root=x;
    while(father[root]!=root)//查找祖先
    {
        root=father[root];
    }
    while(root!=x)路径压缩
    {
        int t=father[x];
        father[x]=root;
        x=t;
    }
}

void Unoin(int x,int y)
{
    int r1=Find(x);
    int r2=Find(y);
    father[r1]=r2;//令x的祖先的父亲是y的祖先
}