并查集(Union Find)
1、查找(Find)
判断两个值是否在同一个集合中。
复杂度为:O(n)↓
int find(int x){
if(father[x] == x){
return x;
}
return father[x];
}
优化代码↓
复杂度:O(1)
int find(int x){
if(father[x] == x){
return x;
}
father[x] = find(father[x]);
return father[x];
}
优化原因:
优化前:当查找某一个值所在的集合时,只是单纯的查找某一值所在的集合。那么当集合嵌套很深的情况,即 A 属于 B 集合,B 集合又属于 C 集合, ……,D 集合又属于 E 集合。那么每次查找的时候都需要遍历所有的节点。
优化后:在查找到某一值所在的集合后,直接把该值指向集合属于的集合,(还用上述的嵌套关系)即在知道 D 属于 E 集合后,把 D 指向 E;在得知 C 集合属于 D 集合后,直接把 C 指向 D 所属的集合即 E 集合……,这样以后每个值只会指向最大的那个集合,若再次执行 find 只需 O(1)。
2、合并(Union)
可以把两个值对应的集合合并。
复杂度为:O(n)↓
void union(int a, int b){
int root_a = find(a);
int root_b = find(b);
if(root_a != root_b){
father[root_a] == root_b;
}
}
经过上述的优化后,合并阶段的复杂度也为O(1).
3、结合后的完整代码
public class UnionFind{
private int[] father = null;
int find(int x){
if(father[x] == x){
return x;
}
father[x] = find(father[x]);
return father[x];
}
void union(int a, int b){
int root_a = find(a);
int root_b = find(b);
if(root_a != root_b){
father[root_a] == root_b;
}
}
}