并查集 数据结构

并查集是一种由若干集合组成的数据结构。它支持合并集合和查询两个数是否在一个集合两种操作。

并查集特性与操作

下面的代码就是朴素的并查集：

int f[100];//并查集数组
int find(int x) {
    if(f[x] == 0) return x;
    return find(f[x]);
}

我们结合代码来看并查集的几个特性：

首先，并查集的初始状态一般为每一个数独自为一个集合。

其次，ff 数组里是一个数，它初始指向自己，ff 数组的含意为：对于任意数 xx, f_xfx​ 为它的直接祖先。若为 0，则它为树根。 每一个树都代表一个集合。

如此一来， 两个操作迎刃而解，若两个数在同一棵树内。则它们在同一集合内。合并同理。

所以，如何判断两个数是否在同一集合内？因为一棵树只有一个树根，所以若它们的树根相同即可。合并时将两个树根合并即可，所以合并和查询的代码如下：

bool in(int a, int b){
    return find(a) == find(b);
}
int change(int a, int b){
    f[find(b)] = a;
}

路径压缩

路径压缩优化，即在查找树根时令 f_ifi​ 直接指向树根。

代码：

int f[100];//并查集数组
int find(int x) {
    if(f[x] == 0) return x;
    return f[x] = find(f[x]);
}