题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定: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<stdio.h>
int fa[20010],n,m,k,a,b,c,d;
int find(int x){
if(x==fa[x]) return x;
return fa[x]=find(fa[x]);
}
void join(int a,int b){
int f1=find(a),f2=find(b);
if(f1!=f2) fa[f1]=f2;
}
int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=m;i++){
scanf("%d%d",&a,&b);
join(a,b);
}
for(int i=1;i<=k;i++){
scanf("%d%d",&c,&d);
printf(find(c)==find(d)?"Yes\n":"No\n");
}
}
用fa数组记录每个人的族长,如果fa[i]==fa[j],那么i和j就是亲戚。