🍄前言
大家好,我是一勺黑猫。今天是每日一题的第十一天,欢迎更多小伙伴加入到我们的打卡计划中,希望和你们在学习算法的路上一起进步~
🙎作者简介:一个正在努力学算法和后端的大三girl
⏳每日一题打卡地:高校算法学习社区
🎈联系方式:157543570(qq)
🍄今日题目
🌰思路:这道题目是并查集的典型应用。并查集的核心是father数组,find函数以及union函数。father数组用于存储每个节点的父节点,find函数用于查找每个节点的祖先节点,union函数用于合并两个关联的节点(将一个节点的父节点指向另一个节点)。
据此,find函数可以写成:
int find(int x){
//如果当前节点的父节点不是祖先节点,就继续向上查找
while(father[x]!=x)
x=father[x];
return x;
}
这样一来,我们每次调用find函数,都要逐个节点查找,比较费时。
由此产生了路径压缩算法,它的主要思想是在逐层查找的过程中,将当前节点到祖先节点之间的所有节点的父节点都指向祖先节点,这样在下次查找其中某个节点的时候,就可以直接返回其祖先节点了。
采用路径压缩算法的代码如下:
int find(int x){
if(father[x]!=x)
father[x]=find(father[x]); //将路径上每个节点的父节点都变成祖先节点
return father[x];
}
有无压缩算法的对比如下,可以发现确实快了很多
🌰AC代码
#include<iostream>
using namespace std;
//存储父节点
int father[5010];
//查找祖先节点
int find(int x){
//路径压缩,将当前节点到祖先节点之间的每个节点的父节点都指向祖先节点
if(father[x]!=x)
father[x]=find(father[x]);
return father[x];
}
//将两个相关联的节点合并
void u(int x,int y){
int xF=find(x),yF=find(y);
if(xF!=yF)
father[xF]=yF; //这里题目没有限制,写成father[yF]=xF也可以
}
int main(){
int n,m,p;
cin>>n>>m>>p;
//初始化父节点数组,让每个节点的父节点是自己(假设一开始都不相关)
for(int i=1;i<=n;i++)
father[i]=i;
int a,b;
for(int i=0;i<m;i++){
cin>>a>>b;
u(a,b); //将具有亲戚关系的两个节点合并
}
for(int i=0;i<p;i++){
cin>>a>>b;
cout<<(find(a)==find(b)?"Yes":"No")<<endl; //判断他们的祖先是否相同
}
return 0;
}