朴素并查集和维护size的并查集

朴素并查集

#include <iostream>

using namespace std;

// http://t.csdn.cn/VZ8gb


/*

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

现在要进行 m 个操作，操作共有两种：
1." M a b"，将编号为 a 和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2." Q a b"，询问编号为 a 和 b 的两个数是否在同一个集合中；

输入格式
第一行输入整数 n 和 m 。
接下来 m 行，每行包含一个操作指令，指令为" M a b "或" Q a b "中的一种。

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

*/





// 并查集
// 1、将两个集合合并
// 2、询问两个元素是否在一个集合中

const int N = 100010;

int p[N]; // 每个元素的父节点

int n, m;



// 返回x的根节点（祖宗节点）
// 已加上路径压缩，即让每个点的父节点为自己的根节点（祖宗节点）
int find(int x)
{
	if (p[x] != x)
	{
		p[x] = find(p[x]);
	}

	return p[x];
}


// 将 a所在集合 合并到 b所在集合
void merge(int a, int b)
{
	p[find(a)] = find(b);
}

int main()
{
	cin >> n >> m;

	// 初始化
	for (int i = 0; i < n; i++)
	{
		p[i] = i;
	}


	// 处理每个操作
	while (m--)
	{
		char op;
		int a, b;

		cin >> op >> a >> b;

		if (op == 'M')
		{
			merge(a, b);
		}

		else if (op == 'Q')
		{
			if (find(a) == find(b))
			{
				cout << "Yes" << endl;
			}
			else
			{
				cout << "No" << endl;
			}
		}
	}


	return 0;
}

维护size的并查集

#include <iostream>

using namespace std;


/*

给定一个包含 n 个点（编号为1 ~ n ）的无向图，初始时图中没有边。

现在要进行 m 个操作，操作共有三种：
1." C a b "，在点 a 和点 b 之间连一条边， a 和 b 可能相等；
2."Q1 a b"，询问点 a 和点 b 是否在同一个连通块中， a 和 b 可能相等；
3."Q2 a"，询问点 a 所在连通块中点的数量；

输入格式
第一行输入整数 n 和 m 。
接下来 m 行，每行包含一个操作指令，指令为" C a b ","Q1 a b "或"Q2 a "中的一种。


输出格式
对于每个询问指令"Q1 a b "，如果 a 和 b 在同一个连通块中，则输出" Yes "，否则输出" No "。
对于每个询问指令"Q2 a "，输出一个整数表示点 a 所在连通块中点的数量
每个结果占一行。

*/




const int N = 100010;

// p存储父节点，sz存储集合元素的个数(只保证根节点的sz是有意义的)
int p[N], sz[N];

int n, m;


int find(int x)
{
	if (p[x] != x)
	{
		p[x] = find(p[x]);
	}

	return p[x];
}

void merge(int a, int b)
{
	// 更新集合元素个数
	sz[find(b)] += sz[find(a)];

	// 将a所在集合并入b所在集合
	p[find(a)] = find(b);
}

int main()
{
	cin >> n >> m;

	for (int i = 1; i <= n; i++)
	{
		p[i] = i;
		sz[i] = 1;
	}

	// 处理询问操作
	while (m--)
	{
		char op[5];
		cin >> op;

		if (op[0] == 'C')
		{
			int a, b;
			cin >> a >> b;

			if (find(a) != find(b))
			{
				merge(a, b);
			}
		}
		else if (op[0] == 'Q')
		{
			if (op[1] == '1')
			{
				int a, b;
				cin >> a >> b;

				if (find(a) == find(b))
				{
					cout << "Yes" << endl;
				}
			}
			else if (op[1] == '2')
			{
				int a;
				cin >> a;

				cout << sz[find(a)] << endl;
			}
		}
	}

	return 0;
}