Prim算法用于计算最小生成树。Prim算法分为两种,一种是懒汉式,一种是饿汉式。
懒汉式Prim算法步骤如下:
首先将顶点0加入到MST中
从MST与未访问顶点之间边中选出最短的边,在满足MST的前提下,将这条边加入到MST
import java.util.LinkedList; import java.util.List; public class LazyPrim { private MinPQ<Edge> pq = new MinPQ<Edge>(); private List<Edge> mst = new LinkedList<Edge>(); private boolean[] visited; public LazyPrim(EdgeWeightedGraph G) { visited = new boolean[G.V()]; // 将第一个顶点加入到MST中 visit(G, 0); // 依次选出最短的边 while (!pq.isEmpty()) { Edge minEdge = pq.popMin(); int v = minEdge.either(); int w = minEdge.other(v); // 忽略造成回路的边 if (visited[v] && visited[w]) { continue; } // 访问该点,将该边加入到MST if (!visited[v]) visit(G, v); else visit(G, w); mst.add(minEdge); } } public Iterable<Edge> edges() { return mst; } public double weight() { double result = 0; for (Edge e : edges()) { result += e.weight(); } return result; } private void visit(EdgeWeightedGraph G, int v) { // 将该点标记为已访问 visited[v] = true; // 将该点伸展的边加入到优先级队列中 for(Edge e:G.adj(v)){ int w = e.other(v); if(!visited[w]) { pq.add(e); } } } }
懒汉式Prim算法在最坏的情况下时间复杂度为E logE,空间复杂度是E。
饿汉式Prim的基本思想就是给每个尚未加入到MST的顶点都只记录一个最短的,连接MST的边。因为只记录一个边,所以计算量稍微减少了一些。
Prim算法中需要用到优先级队列。根据优先级队列的实现方式,算法复杂度也有所差异。如果用二叉堆方式进行时间,那么它的时间复杂度就是E logV,如果用数组方式实现,那么这种算法的复杂度就是V^2。
对于边数非常多的图,使用数组方式时间优先级队列可以提高算法性能,对于边数较少的图,使用二叉堆方式实现优先级队列可以获得较高的性能。
原文地址:http://blog.csdn.net/caipeichao2/article/details/32728597