[YYOJ]亲戚关系[并查集]

题目描述

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。
规定：x和y是亲戚，y和z是亲戚，那么x和z也是亲戚。如果x,y是亲戚，那么x的亲戚都是y的亲戚，y的亲戚也都是x的亲戚。

输入

第一行：三个整数n,m,p，（n<=5000,m<=5000,p<=5000），分别表示有n个人，m个亲戚关系，询问p对亲戚关系。
以下m行：每行两个数Mi，Mj，1<=Mi，Mj<=N，表示Mi和Mj具有亲戚关系。
接下来p行：每行两个数Pi，Pj，询问Pi和Pj是否具有亲戚关系。

输出

P行，每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。

样例输入 Copy

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

样例输出 Copy

Yes
Yes
No

思路

并查集模板
1.初始化init()
2.查找函数
3.合并函数

AC代码

#include <cstdio>
#include <iostream>
using namespace std;
int s[10005];
void init(int x){
	for(int i=1;i<=x;i++){
		s[i]=i;
	}
}
int find(int x){
	if(s[x]==x){
		return x;
	}
	return find(s[x]);

}
void merge(int a,int b){
	int a1=find(a);
	int b1=find(b);
	if(a1!=b1){
		s[a1]=b1;
	}
}
int main(){
	ios::sync_with_stdio(false);
	int n,m,p;
	while(cin>>n>>m>>p){
		init(n);
		for(int i=0;i<m;i++){
			int a,b;
			cin>>a>>b;
			merge(a,b);
		}
		for(int i=0;i<p;i++){
			int k,q;
			cin>>k>>q;
			if(find(k)==find(q)){
				cout<<"Yes"<<endl;
			}else{
				cout<<"No"<<endl;
			}
		}
	}
	return 0;
}