题目
冰岛是作为一个人口稀少的国家,人群之间具有复杂的血缘关系,为了避免不必要的意外,他们的手机上都安装了一款可以随时查询两个人之间是否有血缘关系的软件。现在你的任务就是实现这样一个功能,接收血缘关系的登记信息,并在我们查询时给出两个人是否具有血缘关系。
血缘关系具有自反性、传递性。
输入格式
一行整数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
结尾无空行
数据范围:
对于20%的数据,n<=1000
对于100%的数据,n<=10^7 ,1<=a,b<=n
基本思路
并查集问题,直接套用模板就可以解决。
代码
#include <bits/stdc++.h>
using namespace std;
int n;
vector<int> father;//下标为人物编号,值为该人物的父结点的编号
//查找
int findFather(int v){
if(father[v]==v){
return father[v];
}else{
//路径压缩
int F=findFather(father[v]);
father[v]=F;
return F;
}
}
//合并
void Union(int a,int b){
int faA=findFather(a);
int faB=findFather(b);
if(faA!=faB){
father[faA]=faB;
}
}
int main(){
cin>>n;
father.resize(n+1);
//初始化并查集数组father
for(int i=1;i<=n;i++) father[i]=i;
for(int i=0;i<n;i++){
int q,a,b;
cin>>q>>a>>b;
if(q==1){
Union(a,b);
}else if(q==2){
int faaa=findFather(a);
int fbbb=findFather(b);
// cout<<faaa<<" "<<fbbb<<endl;
if(faaa==fbbb){
cout<<"YES"<<endl;
}else{
cout<<"NO"<<endl;
}
}
}
}