计蒜客 朋友

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/weixin_41282486/article/details/80614194

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

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

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

输入格式 

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

接下来 m 行：每行两个数Ai,Bi(0<=Ai,Bi<=N)表示 Ai和 Bi具有朋友关系。 

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

输出格式 

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

样例输入 

6 5 3 

1 2 

1 5 

3 4 

5 2 

1 3 

1 4 

2 3 

5 6 

样例输出 

Yes 

Yes 

No

#include<stdio.h>
int ans[5050],f[5050],n,m,p;
//初始化 
void init(){
	for(int i = 1;i <= n;i ++){
		f[i] = i;
	}
}
//找根节点，路径压缩 
int get(int x){
	if(f[x] == x){
		return x;
	}
	int y = f[x];
	f[x] = get(y);
	return f[x];
}
//合并 
void merge(int x,int y){
	int n = get(x);
	int m = get(y);
	
	if(n != m){
		f[n] = m;
	}
}

int main(){
	int x,y;
	scanf("%d%d%d",&n,&m,&p);
	init();
	for(int i = 1;i <= m;i ++){
		scanf("%d%d",&x,&y);
		merge(x,y);
	}
	for(int i = 1;i <= p;i ++){
		scanf("%d%d",&x,&y);
		if(get(x) == get(y)){
			ans[i] = 1;
		}
	}
	for(int i = 1;i <= p;i ++){
		if(ans[i] == 1){
			printf("Yes\n");
		}else{
			printf("No\n");
		}
	}
	return 0;
}