并查集

维护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);
}
posted on 2021-11-10 16:34  naiji  阅读(39)  评论(0)    收藏  举报