亲戚 并查集

问题描述：

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。

规定：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");
	}
}