洛谷--[模板]并查集

思路：合并在同一个集合里，即两者有共同祖先。查询就是查找两者的祖先是否相同。

#include <bits/stdc++.h>
using namespace std;
#define MAX 200000
int f[MAX] = { 0 };//f[i]:第i个人的祖先
int find(int m)//寻找m的祖先
{
	if (f[m] == m)
		return m;
	f[m] = find(f[m]);
	return f[m];
}
int main()
{
	int n, m;
	cin >> n >> m;
	for (int i = 0; i < n; i++)
		f[i] = i;//初始化祖先为自己
	int z, x, y;
	for (int i = 0; i < m; i++)
	{
		cin >> z >> x >> y;
		if (z == 1)
			f[find(x)] = find(y);//y的祖先是x祖先的祖先
		else {
			if (find(x) == find(y))
				cout << "Y" << endl;
			else cout << "N" << endl;
		}
	}
}

三个问题：

①find()函数判断语句中，为什么是f[m]== m而不是find(m) == f[m]？

因为在并查集中，每个节点的父节点指向其自己作为根节点，自己是自己的祖先。

因此，在初始状态下，f[m]被初始化为节点m。

如果f[m]的值等于m，则表示该节点m是祖先。

②find()函数中，什么是f[m] = find(f[m])而不是f[m] = find(m)？

通过调用find(f[m])来找到f[m]所属的根节点，并将其赋给f[m]，实现路径压缩。

find(m)会导致递归函数一直调用自身

③主函数中，查询时，为什么是f[find(x)] = find(y)？

y和x要进入一个集合，而x的祖先不是y的祖先，所以让x祖先的祖先变成y的祖先，这样就有共同祖先。