最小生成树Kruskal算法-java实现(使用并查集结构)
克鲁斯卡尔算法百度到的解释是:克鲁斯卡尔算法是一种用来寻找最小生成树的算法。在剩下的所有未选取的边中,找最小边,如果和已选取的边构成回路,则放弃,选取次小边。
1.将图的所有连接线去掉,只剩顶点
2.从图的边集数组中找到权值最小的边,将边的两个顶点连接起来
3.继续寻找权值最小的边,将两个顶点之间连接起来,如果选择的边使得最小生成树出现了环路,则放弃该边,选择权值次小的边
4.直到所有的顶点都被连接在一起并且没有环路,最小生成树就生成了。
具体的过程的解析在代码中都已经详细的注释了,包括并查集结构的使用,这里用到了自己定义的比较器来利用优先级队列实现小根堆,小根堆的作用是根据边的权值的大小进行排序。
代码如下:
import java.util.Collection;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;
//undirected graph only
public class Code_04_Kruskal {
// Union-Find Set
public static class UnionFind {
private HashMap<Node, Node> fatherMap;
private HashMap<Node, Integer> rankMap;
public UnionFind() {
fatherMap = new HashMap<Node, Node>();
rankMap = new HashMap<Node, Integer>();
}
private Node findFather