PTA——冰岛家谱

PTA——冰岛家谱

冰岛是作为一个人口稀少的国家，人群之间具有复杂的血缘关系，为了避免不必要的意外，他们的手机上都安装了一款可以随时查询两个人之间是否有血缘关系的软件。现在你的任务就是实现这样一个功能，接收血缘关系的登记信息，并在我们查询时给出两个人是否具有血缘关系。

血缘关系具有自反性、传递性。

输入格式：

一行整数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<=1e7，1<=a,b<=n

分析：

这题主要是运用并查集来求解，但需要注意的是不能用cin,cout来输入输出，在第三个点会超时。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e7+10;
int vis[maxn];
int find(int x)
{
	if(vis[x]==x){
		return x;
	}
	else{
		vis[x]=find(vis[x]);
		return vis[x];
	}
}
void merge(int a,int b)
{
	int p=find(a);
	int q=find(b);
	if(p==q){
		return ;
	}
	else{
		if(q>p) 
			vis[q]=p;
		else
			vis[p]=q;
		return ;
	}
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=maxn;i++){
		vis[i]=i;
	}
	while(n--){
		int x,y,z;
		scanf("%d %d %d",&x,&y,&z);
		if(x==1){
			merge(y,z);
		}
		else if(x==2){
			if(find(y)==find(z)){
				printf("YES\n");
			}
			else{
				printf("NO\n");
			}
		}
	}
	return 0;
}