每日一题 合并集合

合并集合

一共有 n 个数，编号是 1∼n，最开始每个数各自在一个集合中。

现在要进行 m 个操作，操作共有两种：

M a b，将编号为 a 和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；
输入格式
第一行输入整数 n 和 m。

接下来 m 行，每行包含一个操作指令，指令为 M a b 或 Q a b 中的一种。

输出格式
对于每个询问指令 Q a b，都要输出一个结果，如果 a 和 b 在同一集合内，则输出 Yes，否则输出 No。

每个结果占一行。

数据范围
1≤n,m≤10⁵

输入样例：

4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

输出样例：

Yes
No
Yes

简单明了的了解并查集可以参考这篇文章
分析：
简单的并查集类型的题目。用树来表示同一集合（指向同一根节点的就是同一个集合）。

fa[x] 表示x节点的父节点。初始状态时候，每个元素都是各自树的根。

将两个集合（x，y）合并：将x的根指向y的根（合并两棵树，y指向x也行），x和y在同一根下，属于同一集合。

检查两个集合是否属于同一集合：检查两元素的根是否相同。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 100005;
int fa[N];
int n, m;
int get(int x) //检查x的根
{
	if (x == fa[x]) return x;
	return fa[x] = get(fa[x]);
}
void merge(int a, int b) //合并a和b集合
{
	int x = get(a), y = get(b);
	fa[y] = x;
}
int main()
{
	cin >> n >> m;
	for (int i = 1;i <= n;i++) fa[i] = i;

	int a, b;
	char flag;
	while (m--)
	{
		cin >> flag >> a >> b;
		if (flag == 'M') merge(a, b);
		else if(flag=='Q')
		{
			if (get(a) == get(b)) cout << "Yes" << endl;
			else cout << "No" << endl;
		}
	}
	return 0;
}