问题描述
最小生成树的概念:
最小生成树的意思就是,给你很多边和点的图,请你求出使得所有顶点都能联通起来,同时边的权值之和最小的结果,这个结果就是最小生成树。
思路分析
既然要求把所有顶点都连起来,并且边的权值最小,那么,一般就有两种思路,一种是从边出发,每次挑选最小的边,这是克鲁斯卡尔算法,还有一种就是普利姆算法了,从顶点出发,每次挑选出当前可以走的最短路径。
普利姆算法思路详解:
首先,从任意一个节点开始,然后,把当前节点的所有边加入到一个集合:解锁集合。
如何理解解锁集合?解锁集合的意思就是,当前可供选择的所有边,然后,将解锁集合排序,每次从中选取最小的边,然后观察最小的边连接的节点是否已经被处理过,如果被处理过,就继续选择下一个最小的边,如果没被处理过,那么,就相当于发现一个新节点,把新节点加入结果集中,然后,将新节点所有边再加入解锁集合,然后依次重复上述步骤,直到解锁集合遍历完,仍然未发现新节点,说明最小生成树完成了。
代码
public static Set<Edges> primMST(Graph graph){
//优先级队列
PriorityQueue<Edges> queue=new PriorityQueue<>(new EdgeComparator());
//保存结果
HashSet<Edges> result=new HashSet<>();
//判断某个节点是否已经遍历过
HashSet<Node> set=new HashSet<>();
//外层循环解决森林的问题
for (Node n : graph.nodes.values()){
if(!set.contains(n)){
for (Edges edge : n.edges) {
queue.add(edge);
}
set.add(n);
while (!queue.isEmpty()){
Edges e=queue.poll();
Node to=e.to;
if(!set.contains(to)){
result.add(e);
set.add(to);
for (Edges edges : to.edges){
queue.add(edges);
}
}
}
}
}
return result;
}