【模板】并查集

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式

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

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

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

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

输出格式

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

输入输出样例

输入 #1

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

输出 #1

N
Y
N
Y

说明/提示

对于 30%30% 的数据，N≤10，M≤20。

对于 70%70% 的数据，N≤100，M≤103。

对于 100%100% 的数据，1≤N≤104，1≤M≤2×105，1≤Xi​,Yi​≤N，Zi​∈{1,2}。

代码如下：

#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(){
	cin>>n>>m;
	init();
	for(int i=0;i<m;i++){
		int z,x,y;
		cin>>z>>x>>y;
		if(z==1){
			mer(x,y);
		}else if(z==2){
			if(check(x,y)) cout<<"Y"<<endl;
			else cout<<"N"<<endl;
		}
	}
	return 0;
}