题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定:x 和 y 是亲戚,y 和 z 是亲戚,那么 x 和 z 也是亲戚。如果 x,y 是亲戚,那么 x 的亲戚都是 y 的亲戚,y 的亲戚也都是 x 的亲戚。
输入格式
第一行:三个整数 n,m,p,(n,m,p≤5000),分别表示有 n 个人,m 个亲戚关系,询问 p 对亲戚关系。
以下 m 行:每行两个数 Mi,Mj,1≤Mi, Mj≤N,表示 Mi 和 Mj 具有亲戚关系。
接下来 p 行:每行两个数 Pi,Pj,询问 Pi 和 Pj 是否具有亲戚关系。
输出格式
p 行,每行一个 Yes
或 No
。表示第 i 个询问的答案为“具有”或“不具有”亲戚关系。
输入输出样例
输入 #1
6 5 3 1 2 1 5 3 4 5 2 1 3 1 4 2 3 5 6
输出 #1
Yes Yes No
代码如下:
#include<bits/stdc++.h>
using namespace std;
int n,m;
vector<int> f;
void init(){
f.resize(n+1);
for(int i=0;i<=n;i++) f[i]=i;
}
int fa(int a){
if(f[a]==a) return a;
return f[a]=fa(f[a]);
}
bool check(int u,int v){
u=fa(u);v=fa(v);
return u==v;
}
void mer(int u,int v){
u=fa(u);v=fa(v);
f[u]=v;
}
int main(){
int z,x,y;
cin>>n>>m>>z;
init();
for(int i=0;i<m;i++){
cin>>x>>y;
mer(x,y);
}
for(int i=0;i<z;i++){
cin>>x>>y;
if(check(x,y)) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
return 0;
}