学习笔记（4）并查集

前两天听蓝桥杯讲座的时候听到了这个记忆中尘封已久的词汇，就琢磨着预习复习一下这个知识点。

---

对于并查集中的每一个元素，都只有一个父亲节点，显然并查集是一种树型的数据结构，用于处理一些不相交集合的合并问题。

并查集的主要操作有： 1：确定元素属于哪一个集合

                                     2：将两个不相交子集合并成一个集合

并查集介绍

一、初始化方法如下：

int f[10000];
void init()
{
    for(int i=0;i<10000;i++)
    f[i]=i;//初始化默认把自己设为父亲节点
}

二、寻找父亲节点（两种路径压缩方法）

1、循环法：

int find(int x)
{

    while(x != f[x]){//如果x元素的父亲指针指向的不是自己，说明x并不是集合中的根元素，还需要一直向上查找和路径压缩
                //在find查询中嵌入一个路径压缩操作
                f[x]=f[f[x]];
                //x元素不再选择原来的父亲节点，而是直接选择父亲节点的父亲节点来做为自己新的一个父亲节点
                //这样的操作使得树的层数被压缩了
                x=f[x];//x压缩完毕后且x并不是根节点，x变成x新的父节点继续进行查找和压缩的同时操作
            }
            return x;//经过while循环后，x=f[x],一定是一个根节点，且不能够再进行压缩了,我们返回即可
}

 2、递归法：

int find(int x)
{
    if(x==f[x]) {//此时x为根节点
    return x;}//返回该结合的根节点
    else{//此时遍历的不为根节点
    f[x]=find(f[x]);//通过递归条用find（f[x])来向上访问元素，并且把找到的根元素赋值给节点的父亲指针
    return f[x];}//返回此时访问的节点的父亲指针指向的节点，也就是这个集合的根节点

}

 

三、并操作（Union）

void Union(int a,int b)
{
    int fa=find(a);//找出a元素位于集合的根元素
    int fb=find(b);//找出b元素位于集合的根元素
    if(fa!=fb)     //如果两个根元素不同
    f[fa] = fb;    //则将两个集合合并
}

 并查集实际应用

 【模板】并查集 - 洛谷

这题直接上模板

#include<bits/stdc++.h>
using namespace std;

int A[10050];
int f[10050];
void init(){
    for(int i=0;i<10050;i++)
    f[i] = i;
}
int find(int x)
{
    if(f[x]==x)return x;
    else
    {
        f[x]=find(f[x]);
        return f[x];
    }
}

void Union(int u,int v)
{
    int fu=find(u);
    int fv=find(v);
    if(fu!=fv)
    f[fu] = fv;
}
int main()
{
    init();
    int n,m;
    cin>>n>>m;
    while(m--)
    {
        int z,x,y;
        cin>>z>>x>>y;
        if(z==1)
        {
            Union(x,y);
        }
        else{
            if(find(x)==find(y))cout<<"Y"<<endl;
            else cout<<"N"<<endl;
        }
    }
}

亲戚 - 洛谷

这题也是

#include<bits/stdc++.h>
using namespace std;
int f[10000];
void init()
{
    for(int i=0;i<10000;i++)
    f[i]=i;
}
int find(int x)
{
    if(f[x]==x)return x;
    else
    {
        f[x]=find(f[x]);
        return f[x];
    }
}
void Union(int a,int b)
{
    int fa=find(a);//找出a元素位于集合的根元素
    int fb=find(b);
    if(fa!=fb)
    f[fa]=fb;
}
int main()
{
    init();
    int n,m,p;
    cin>>n>>m>>p;
    while(m--)
    {
        int i,j;
        cin>>i>>j;
        Union(i,j);
    }
    while(p--)
    {
        int i,j;
        cin>>i>>j;
        if(find(i)==find(j))cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
}

修复公路 - 洛谷

这题不是很简单，说一下思路

首先对于N个村庄，倘若能够全部连接，那么最后就只剩下一个集合。

所以，每一次读取某条公路连接两个村庄时，需要判断一下父亲节点是否相同。

如果不同，则将它们并起来且单独的村庄数量-1。

由于每条公路修成的时间不同，所以我们按每条路的修建时间来连通村庄。（用到了sort的结构体排序）

整体代码如下：

#include<bits/stdc++.h>
using namespace std;
int f[10000];
void init()
{
    for(int i=0;i<10000;i++)
    f[i]=i;
}
int find(int x)
{
    if(f[x]==x)return x;
    else
    {
        f[x]=find(f[x]);
        return f[x];
    }
}
void Union(int a,int b)
{
    int fa=find(a);//找出a元素位于集合的根元素
    int fb=find(b);
    if(fa!=fb)
    f[fa]=fb;
}
typedef struct a{
    int x,y,t;
}g;
bool cmp(g a,g b)
{
    return a.t<b.t;
}
int main()
{
    init();
    int n,m;
    cin>>n>>m;
    g a[100500];
    for(int i=1;i<=m;i++)
    {
        cin>>a[i].x>>a[i].y>>a[i].t;
    }
    sort(a+1,a+1+m,cmp);

    for(int i=1;i<=m;i++)
    {
        int fx=find(a[i].x);//找出a元素位于集合的根元素
        int fy=find(a[i].y);
        if(fx!=fy)
        {f[fx]=fy;
        n--;}
        if(n==1){cout<<a[i].t;return 0;}
    }
    cout<<-1<<endl;

}

参考：并查集的路径压缩优化_扬俊的小屋-CSDN博客_并查集路径压缩

参考：并查集详解_独钓烟云的博客-CSDN博客