【模板】并查集

【模板】并查集

题目描述

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

输入格式

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

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

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

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

输出格式

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

---

裸的冰茶姬模板，没什么好说的，写一遍以便复习。

个人习惯在使用合并操作时套一个查询操作，看个人爱好啦~

加了一个最简单的优化
 

#include <iostream>
using namespace std;
int f[10010];//父亲节点
int get_root(int a)
{
	if(f[a]==a) return a;
	return f[a]=get_root(f[a]);//路径压缩
}
void merge(int n,int m)
{
	int a=get_root(n);
	int b=get_root(m);
	if(a!=b) f[a]=b;//合并两个集合
}
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++) f[i]=i;//初始化
	for(int i=1;i<=m;i++)
	{
		int z,x,y;
		cin>>z>>x>>y;
		if(z==1) merge(x,y);//合并操作
		else//判断是否处于一个集合中
		{
			int a=get_root(x);
			int b=get_root(y);
			if(a==b) cout<<"Y"<<endl;
			else cout<<"N"<<endl;
		}
	}
	return 0;
}