问题描述:
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。(人数≤5000,亲戚关系≤5000,询问亲戚关系次数≤5000)。
数据输入:第一行:三个整数n,m,p,(n<=5000,m<=5000,p<=5000),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。
以下m行:每行两个数Mi,Mj,1<=Mi,Mj<=N,表示Ai和Bi具有亲戚关系。
接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。
数据输出:
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系
样例:
input
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
output
Yes
Yes
No
#include<stdio.h>
#include<iostream>
using namespace std;
int fa[50005],sum[50005];
int find(int a){//查找该集合中的根
if(fa[a]==a) return a;
else{
int root=find(fa[a]);
fa[a]=root;//深度压缩
return root;
}
}
int main(){
int n,m,p,a,b,i;
scanf("%d%d%d",&n,&m,&p);
for(i=0;i<n;i++){
fa[i]=i;
sum[i]=0;
}
for(i=0;i<m;i++){
scanf("%d%d",&a,&b);
if(find(a)!=find(b)){//在不同集合就合并
if(sum[a]>=sum[b]){// 小集合并到大集合里
fa[b]=find(a);//b的根变成a集合的根
sum[a]+=sum[b];//a集合的元素变多
}
else{
fa[a]=find(b);
sum[b]+=sum[a];
}
}
}
for(i=0;i<p;i++){
scanf("%d%d",&a,&b);
if(find(a)==find(b)){
printf("Yes\n");
}
else printf("No\n");
}
}