蓝桥杯：朋友

在社交的过程中，通过朋友，也能认识新的朋友。在某个朋友关系图中，假定 A 和 B 是朋友，B 和 C 是朋友，那么 A 和 C 也会成为朋友。即，我们规定朋友的朋友也是朋友。

现在，已知若干对朋友关系，询问某两个人是不是朋友。

请编写一个程序来解决这个问题吧。

输入格式

第一行：三个整数 n,m,p(n≤5000,m≤5000,p≤5000)，分别表示有n个人，m个朋友关系，询问p对朋友关系。

接下来 m 行：每行两个数 Ai,Bi，1≤Ai,Bi≤N，表示 Ai和Bi具有朋友关系。

接下来p行：每行两个数，询问两人是否为朋友。

输出格式

输出共p行，每行一个Yes或No。表示第i个询问的答案为是否朋友。

样例输入

6 5 3

1 2

1 5

3 4

5 2

1 3

1 4

2 3

5 6

样例输出

Yes

Yes

  No

 

#include <iostream>
using namespace std;

int father[5002];

int get(int x){
	if(father[x]==x) return x;
	
	int y=father[x];
	father[x]=get(y);
	
	return father[x];
}
void merge(int a,int b){
	a=get(a);
	b=get(b);
	if(a!=b){
		father[a]=b;
	}
}

int main(){
	int n,m,p;
	
	cin>>n>>m>>p;
	/*初始化，每个人都是自己的根节点*/
	for(int i=1;i<=n;i++){
		father[i]=i;
	}
	/*合并有关系的集合*/
	while(m--){
		int a,b;
		cin>>a>>b;
		merge(a,b);
	}
	/*查询每行两个人是否是一个根节点*/
	while(p--){
		int a,b;
		cin>>a>>b;
		if(get(a)==get(b)){
			cout<<"Yes"<<endl;
		}else{
			cout<<"No"<<endl;
		}
	}

	return 0;
}