1.初始化
for(int i=1;i<=n;i++)
a[i]=i;
2.查询
int find(int i){
if(a[i]==i)
return i;
else
return find(a[i]);
}
3.合并
void he(int i,int j){
int i_a=find(i);
int j_a=find(j);
a[i_a]=j_a;
}
上例题
洛谷P1551 亲戚
题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定:xx 和 yy 是亲戚,yy 和 zz 是亲戚,那么 xx 和 zz 也是亲戚。如果 xx,yy 是亲戚,那么 xx 的亲戚都是 yy 的亲戚,yy 的亲戚也都是 xx 的亲戚。
输入格式
第一行:三个整数 n,m,pn,m,p,(n,m,p \le 5000n,m,p≤5000),分别表示有 nn 个人,mm 个亲戚关系,询问 pp 对亲戚关系。
以下 mm 行:每行两个数 M_iMi,M_jMj,1 \le M_i,~M_j\le N1≤Mi, Mj≤N,表示 M_iMi 和 M_jMj 具有亲戚关系。
接下来 pp 行:每行两个数 P_i,P_jPi,Pj,询问 P_iPi 和 P_jPj 是否具有亲戚关系。
输出格式
pp 行,每行一个 Yes
或 No
。表示第 ii 个询问的答案为“具有”或“不具有”亲戚关系。
输入输出样例
输入 #1复制
6 5 3 1 2 1 5 3 4 5 2 1 3 1 4 2 3 5 6
输出 #1复制
Yes Yes No
#include<bits/stdc++.h>
using namespace std;
int n,m,q,f[10010],c,d,a,b;
int fd(int x)//找出x家的大佬 也就是二叉树的祖先节点
{
if(f[x]==x)//x是x的爸爸,简单的来说就是x没爸爸了
//他是家里最大的大佬,所以返回的x就是我们所求的祖先节点
return x;
else
return f[x]=fd(f[x]);//x不是他自己的爸爸,所以他上面还
//有爸爸,我们的目标是祖先节点,所以我们此时要做的是问他
//爸爸的爸爸是谁,即再使用一次fd(find)函数【其实就是一个递归过程
}
void hb(int x,int y)
{
f[fd(y)]=fd(x);//合并x子集和y子集,直接把x子集的祖先节
//点与y子集的祖先节点连接起来,通俗点来说就是把x的最大祖
//先变成y子集最大祖先的爸爸
return ;
}
int main()
{
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=n;i++)
f[i]=i;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&c,&d);
hb(c,d);
}
for(int i=1;i<=q;i++)
{
scanf("%d%d",&a,&b);
if(fd(a)==fd(b))//如果a所在子集的大佬[前面已经解释过了]和b所在子集的大佬一样,即可知a和b在同一个集合
printf("Yes\n");
else
printf("No\n");
}
return 0;
}