并查集的作用
1.将元素分类,生成划分联通分量
2.集合的合并
3.判断各个元素是否属于同一个集合
引入例题
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数 N,MN,M ,表示共有 NN 个元素和 MM 个操作。
接下来 MM 行,每行包含三个整数 Zi,Xi,Yi,Zi。
当Zi=1时,将Xi与Yi所在集合合并。
当Zi=2时,输出Xi与Yi是否在同一集合内,是的输出Y;否则输出N。
输出格式
对于每一个 Zi =2 的操作,都有一行输出,每行包含一个大写字母,为 Y 或者 N 。
输入输出样例
输入:
4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4
输出
N
Y
N
Y
标准的板子题
代码如下
#include<bits/stdc++.h>
using namespace std;
const int X = 10010;
int f[X];
int n,m;
void init(){
for(int i = 0;i <= X - 1;i++){
f[i] = i;
}
}
int find_f(int x){
if(x != f[x]){
return f[x] = find_f(f[x]);
}
return f[x];
}
void join(int x,int y){
int fx = find_f(x),fy = find_f(y);
if(fx != fy){
f[fy] = fx;
}
}
int main(){
init();
cin>>n>>m;
while(m--){
int t,x,y;
cin>>t>>x>>y;
switch(t){
case 1:
join(x,y);
break;
case 2:
find_f(x) == find_f(y)?cout<<"Y"<<endl:cout<<"N"<<endl;
break;
}
}
return 0;
}
解析
分为四个部分:
1.初始化部分,并查集的思想是先让每个元素的祖宗都是自己(树),之后再根据需求合并,查找
void init(){
for(int i = 0;i <= X - 1;i++){
f[i] = i;
}
2.查找部分,并查集是通过数组和对应下标来反映元素与其祖宗的关系,这个函数可以查找元素的祖宗
int find_f(int x){
if(x != f[x]){
return f[x] = find_f(f[x]);
}
return f[x];
}
3.合并部分,并查集在有的时候需要将两个元素合并,归到一类,同样,两个集合合并也只需要将他们的祖宗合并
void join(int x,int y){
int fx = find_f(x),fy = find_f(y);
if(fx != fy){
f[fy] = fx;
}
4.主函数,这个根据各个题目不一样而不同,只能说是随机应变
int main(){
init();
cin>>n>>m;
while(m--){
int t,x,y;
cin>>t>>x>>y;
switch(t){
case 1:
join(x,y);
break;
case 2:
find_f(x) == find_f(y)?cout<<"Y"<<endl:cout<<"N"<<endl;
break;
}
}
return 0;
}
这个是本题的解,值得一提的是在并查集的题目中经常会涉及集合数量的判断,可以用到以下语句
for(int count=0,i=1;i<=n;i++)
if(b[i]==i) count++; //如果他的祖宗对于自己,那么他就是祖宗,代表有一个集合
可能还会涉及是否存在闭合回路的判断:每当加入一条边,如果接入的两个点已经接入了之前的边,那么就存在回路
最后附上并查集+dfs的一道例题
https://blog.csdn.net/qq_45530271/article/details/109189978(转载)