[P3367]【模板】并查集.题解

这橙题无非只是模板题，套并查集模板AC！ 并查集的算法思想十分值得大家学习！

老规矩，先看题目。

题目告诉我们要对$N$个元素进行$M$次操作，对其中的一些进行合并操作，并输出两个集合的合并情况。这里我们首先会想到用并查集！没错，但萌新们不用怕，我接下来讲解一下并查集的思路，你们就明白了！

---

思路

并查集：树形数据结构

听起来比较简单，实现起来也很简单。我们分$3$步学会模板！

首先我们先要学会构建$find$函数(查找一个结点的根结点)
拿这题举例，我们通过一个数组$f[10010]$(开小容易$RE$或$TLE$)。如果其祖先等于自己，直接$return$回去。否则通过递归不断循环，最后找出结点的根结点。
$OK$~那我们接下来就看看$find$函数的定义吧！

int find(int x){
    if(f[x]==x) return x;//如果结点本身就是根结点，直接返回自己
    else return f[x]=find(f[x]);//否则通过路径压缩递归返回根结点
}

关于$find$函数还有个路径压缩的知识点：

else return find(f[x]);//递归返回，一个个找它的父结点，最终找到根节点

else return f[x]=find(f[x]);//直接将结点连到其根节点上。find定义代码最后一句就是它

第二个代码优化时间的方法就叫路径压缩。

我们看两张图就理解了！
如图，求这棵树所有结点的祖先，时间复杂度取决于这棵树的深度，是$O(n^2)$。

如图，求这棵树所有结点的祖先，时间复杂度取决于结点的个数，是$O(n)$。
∴路径压缩能优化时间复杂度。

接下来来是并，我们合并两个结点的根结点时需要用到查($find$函数)！
那我们就看看$join$函数的定义吧！

void join(int x,int y){
	f[find(x)]=find(y);//f数组用来存结点的祖先，通过find函数，合并两个结点x、y的根结点
	return;//合并后别忘了直接返回
}

最后一步直接判断两个结点所在集合是否有共同根结点。用$find$函数就可以了！
那我们再看看判断代码吧！

if(find(a)==find(b))
	cout<<"Y"<<endl;//如果结点a和b有相同的根结点，则输出“Y”
else
	cout<<"N"<<endl;//否则输出“N”

好了，我们的思路就讲到这里，下面是$AC$代码！

---

代码

(注释版)

#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;
}

---

总结

并查集分$3$步：

1. 把每个结点所在的集合初始化为它自己
2. 将两个不同集合的结点合并($join$函数)
3. 查找结点所在集合的根结点($find$函数)