并查集【模板】

最新推荐文章于 2024-09-14 22:25:23 发布

Integricode26

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量131

点赞数 1

分类专栏：算法总结文章标签：算法数据结构

本文链接：https://blog.csdn.net/Integricode26/article/details/107479618

版权

算法总结专栏收录该内容

3 篇文章 0 订阅

订阅专栏

代码

(注释版)

#include<bits/stdc++.h>
using namespace std;
int n,m,f[10010],a,b,c;//数组f[i]记录i的根结点
int find(int x){
    if(f[x]==x) return x;//如果结点本身就是根结点，直接返回自己
    else return f[x]=find(f[x]);//否则通过路径压缩递归返回根结点
}
void join(int x,int y){
	f[find(x)]=find(y);//通过find函数，合并两个结点x、y的根结点
	return;//合并后别忘了直接返回
}
int main(){//好习惯，直接从主函数读起
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        f[i]=i;//初始化根结点数组，确保每个结点现在的根结点都是自己
    for(int i=1;i<=m;i++){
        cin>>a>>b>>c;
        if(a==1)//a为1时进行合并操作
            join(b,c);//直接调用join()函数
        else
            if(find(b)==find(c))//当结点b、c的根结点相等时，则它们在一个集合里
                cout<<"Y"<<endl;
            else//否则它们不属于同一个集合
                cout<<"N"<<endl;
    }
    return 0;
}

(无注释版)

#include<bits/stdc++.h>
using namespace std;
int n,m,f[10010],a,b,c;
int find(int x){
    if(f[x]==x) return x;
    else return f[x]=find(f[x]);
}
void join(int x,int y){
	f[find(x)]=find(y);
	return;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=m;i++){
        cin>>a>>b>>c;
        if(a==1)
            join(b,c);
        else
            if(find(b)==find(c))
                cout<<"Y"<<endl;
            else
                cout<<"N"<<endl;
    }
    return 0;
}