并查集
维护n个元素,刚开始每个元素自己一个集合,支持两个操作。
-
合并两个元素所在的集合
-
询问两个元素是否在相同的集合内
一,带路径压缩的查找函数
int find(int x) {
if (f[x] != x) {
f[x] = find(f[x]);
}
return f[x];
}
二,合并两个元素所在的集合
void merge(int x, int y) {
x = find(x);
y = find(y);
if (x != y) {
f[x] = y;
}
}
合并完成后,若要使所有节点(父节点除外)都压缩到父节点的儿子节点,
则,遍历一遍
for (int i = 1; i <= n; i++)
{
find(i);
}