冰岛是作为一个人口稀少的国家,人群之间具有复杂的血缘关系,为了避免不必要的意外,他们的手机上都安装了一款可以随时查询两个人之间是否有血缘关系的软件。现在你的任务就是实现这样一个功能,接收血缘关系的登记信息,并在我们查询时给出两个人是否具有血缘关系。
血缘关系具有自反性、传递性。
输入格式:
一行整数n,以下n行,每行3个正整数q、a、b
若q为1,则登记a与b具有血缘关系
若q为2,则查询a与b的血缘关系,若有,输出一行YES;若无,输出一行NO。
输出格式:
对于每个q=2,输出一行YES或NO。
样例">样例">输入样例:
5
1 1 2
1 3 4
2 1 3
1 2 3
2 1 4
输出样例:
NO
YES
#include<bits/stdc++.h>
using namespace std;
const int N=10000000;
int n,m;
int f[N];//数组的关系
void init(){
for(int i=1;i<=n;i++){
f[i]=i;
}
}
int find(int x){
if(x!=f[x])f[x]=find(f[x]);
return f[x];
}
void b(int x,int y){
x=find(x);
y=find(y);
if(x!=y) f[x]=f[y];
}
int main(){
int x,y,z;
cin>>n;
init();
while(n--){
cin>>x>>y>>z;
if(x==1){
b(y,z);
}else{
if(find(y)==find(z)){
cout<<"YES"<<endl;
}else{
cout<<"NO"<<endl;
}
}
}
return 0;
}