在社交的过程中,通过朋友,也能认识新的朋友。在某个朋友关系图中,假定 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;
}