此篇为此专栏的必读篇,因为此专栏中所有的并查集代码均使用此篇的并查集代码
构造方法:初始化每个根节点,使每个子集节点数为1
find方法:使用路径压缩,当查找的值不是根节点时,我们把这个值挂到根节点下面
unite方法:使用按秩优化,两个节点传进来,判断是不是同一个根节点,
如果不是同一个根节点,那么我们按照两个节点的节点数来排序,
把节点数小的节点挂到节点数大的节点上,
再让节点数大的节点数量加上节点数小的节点数量(把节点数合并)
getCount方法:判断此时的联通分量的数量
connected方法:判断两个节点是不是同一个根节点
public class UnionFind {
//记录每个节点的根节点
int[] parent;
//记录每个子集的节点数
int[] rank;
//记录并查集中的联通分量数量
int count;
public UnionFind(int n){
count=n;
parent=new int[n];
for (int i=0;i<n;i++){
parent[i]=i;
}
rank=new int[n];
Arrays.fill(rank,1);
}
public int find(int ind){
if (parent[ind]!=ind){
parent[ind]=find(parent[ind]);
}
return parent[ind];
}
public void unite(int ind1,int ind2){
int root1=find(ind1),root2=find(ind2);
if (root1!=root2){
if (rank[root1]<rank[root2]){
int temp=root2;
root2=root1;
root1=temp;
}
parent[root2]=root1;
rank[root1]+=rank[root2];
count--;
}
}
public int getCount(){
return count;
}
public boolean connected(int ind1,int ind2){
return find(ind1)==find(ind2);
}
}
如果您还有什么疑问或解答有问题,可在下方评论,我会及时回复。
系列持续更新中,点个订阅吧