7-51 并查集【模板】 (30 分)

7-51 并查集【模板】 (30 分)

给出一个并查集，请完成合并和查询操作。

输入格式:

第一行包含两个整数N、M，表示共有N个元素和M个操作。

接下来M行，每行包含三个整数Zi​、Xi​、Yi​。

当Zi​=1时，将Xi​与Yi​所在的集合合并。

当Zi​=2时，输出Xi​与Yi​是否在同一集合内，是的话输出Y；否则的话输出N。

输出格式:

对于每一个Zi​=2的操作，对应一行输出，每行包含一个大写字母，为Y或者N。

输入样例:

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

输出样例:

N
Y
N
Y

数据规模:

对于30%的数据，N<=10，M<=20；

对于70%的数据，N<=100，M<=1000；

对于100%的数据，N<=10000，M<=200000。

 

#include<bits/stdc++.h>
using namespace std;

int father[99999];
int n,m;
int find(int a) {
// 	while(a!=father[a])
// 		a=father[a];
// 	return a;//利用循环也会超时
	if(a==father[a])
		return a;
	else
		return father[a]=find(father[a]);//利用递归不会超时
}
void Union(int a,int b) {
	int fa=find(a);
	int fb=find(b);
	if(fa!=fb) father[fa]=fb;
}

int main() {
	//cin>>n>>m;
	scanf("%d%d",&n,&m);
	for(int i=0; i<=n; i++) {
		father[i]=i;
	}
	int x,y,z;
	for(int i=0; i<m; i++) {
		//cin>>x>>y>>z;
		scanf("%d%d%d",&x,&y,&z);
		if(x==1) {
			Union(y,z);
		} else if(x==2) {
			int fy=find(y);
			int fz=find(z);
			if(fy==fz) //cout<<'Y'<<endl;
				printf("Y\n");
			else //cout<<'N'<<endl;
				printf("N\n");
		}
	}
	return 0;
}