并查集
并查集是一种用于维护集合的数据结构,它支持合并集合和查询元素所属集合的操作。并查集常用于解决图论问题,如判断两个节点是否在同一个连通分量中、求图的最小生成树等。
并查集的实现通常使用数组来表示集合,其中每个元素都有一个父节点,最终形成一棵树。树的根节点表示集合的代表元素,树的深度称为秩(rank),用于优化合并操作的效率。
并查集的两个主要操作是查找(find)和合并(union)。
查找操作用于查找元素所属的集合,它通过递归地查找父节点,直到找到根节点为止。在查找时,可以使用路径压缩的技巧将路径上的所有节点直接连接到根节点上,以减少之后的查找时间。
合并操作用于将两个不同的集合合并成一个集合,它的实现通常采用按秩合并的优化策略,即将秩较小的集合合并到秩较大的集合中,以保持树的平衡。
下面是一个使用Java实现的并查集示例
class UnionFind {
private int[] parent;
private int[] rank;
public UnionFind(int n) {
parent = new int[n];
for (int i = 0; i < n; i++) {
parent[i] = i;
}
rank = new int[n];
}
public int find(int x) {
if (parent[x] != x) {
parent[x] = find(parent[x]);
}
return parent[x];
}
public boolean union(int x, int y) {
int px = find(x);
int py = find(y);
if (px == py) {
return false;
}
if (rank[px] < rank[py]) {
parent[px] = py;
} else if (rank[px] > rank[py]) {
parent[py] = px;
} else {
parent[px] = py;
rank[py]++;
}
return true;
}
}
在上面的代码中,我们使用两个数组来表示并查集,其中parent数组表示每个节点的父节点,rank数组表示每个集合的秩。
find方法用于查找元素所属的集合,它通过递归地查找父节点,直到找到根节点为止。在查找时,我们使用路径压缩的技巧将路径上的所有节点直接连接到根节点上。
union方法用于将两个不同的集合合并成一个集合,它的实现采用按秩合并的优化策略,即将秩较小的集合合并到秩较大的集合中,以保持树的平衡。如果两个元素已经在同一个集合中,则合并操作失败,返回false;否则,合并操作成功,返回true。