AcWing836 合并集合 并查集

#include<iostream>
using namespace std;
const int N = 100010;
int fa[N];//定义父节点 
//合并两个集合
//询问两个元素是否在一个集合当中
int find(int x) {
	if (fa[x] != x)fa[x] = find(fa[x]);
	//如果当前这个点不是根节点 那就向上查找 
	//同时压缩路径 使每个点直接指向根节点
	return fa[x];
}
int main() {
	//初始化 依据题意 刚开始每个点的根节点都是自己 即px=x
	int n, m;
	cin >> n >> m;

	for (int i = 1; i <= n; i++) fa[i] = i;
	int a, b;
	string s;
	while (m--) {
		cin >> s >> a >> b ;
		if (s == "M") {
			if (find(a) != find(b))fa[find(a)] = find(b);
			//如果二者的根节点不一致说明不在同一个集合中 
			//那么就让b的根节点成为a的根节点的父节点 即把a连接到b下
		}
		else {
			if (find(a) == find(b))cout << "Yes" << endl;
			else cout << "No" << endl;
		}
	}
	return 0;
}