并查集可以很好的解决连接问题和数学中集合类的实现。
根据两个点的根节点是否相同来判断两个点是否相连。
为了降低树的高度来降低查询复杂度,在合并两个节点时需要判断树的高度,使高度低的指向高度高的,也可以使用路径优化,使所有节点指向同一个父节点。
并查集的操作时间复杂度是O(1)
public class UF {
private int[] parent;
private int[] rank;//用层数来决定父节点
private int count;
public UF(int count){
this.count = count;
parent = new int[count];
rank = new int[count];
for(int i = 0;i<count;i++){
parent[i] = i;
rank[i] = 1;
}
}
public int find(int p){
try {
if(p!=parent[p]){//路径压缩,使所有节点都指向根节点
parent[p] = find(parent[p]);
p = parent[p];
}
} catch (Exception e) {
e.printStackTrace();
}
return parent[p];
}
public boolean isConnected(int p,int q){
return parent[p]==parent[q];
}
public void unionElement(int p,int q){
int pRoot = find(p);
int qRoot = find(q);
if(pRoot==qRoot)
return;
if(rank[pRoot]<rank[qRoot]){
parent[pRoot] = qRoot;
}
else if(rank[pRoot]>rank[qRoot]){
parent[qRoot] = pRoot;
}
else{
parent[pRoot] = qRoot;
rank[qRoot]+=1;
}
}
}