最小生成树--Prim算法

221 阅读1分钟

条件:无向图

任意一个点作为开端,将该点的所有边加到优先级队列,表示边被解锁,从队列中弹出一条边(必是权重最小的)看该边是否已经标记,如果没标记就标记,表示这条边是需要的,然后将该边的相连的另一个点相连的所有边解锁,,,如此类推

图的表示和生成见:点击打开链接

import java.util.Comparator;
import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;

public class primMST {
	public static class EdgeComparator implements Comparator<Edge> {
		@Override
		public int compare(Edge o1, Edge o2) {
			return o1.weight - o2.weight;
		}
	}

	public static Set<Edge> prime(Graph graph) {
		PriorityQueue<Edge> pQueue = new PriorityQueue<Edge>(new EdgeComparator());
		HashSet<Edge> edges = new HashSet<Edge>();
		HashSet<Node> nodes = new HashSet<Node>();
		for (Node node : graph.nodes.values()) {// 深林,要是有不同的树也可以处理
			// node:当前节点,还没进过set
			if (!nodes.contains(node)) {
				// 加进set
				nodes.add(node);
				// node的所有边进优先级队列
				for (Edge curEdge : node.edges) {
					pQueue.add(curEdge);
				}
				while (!pQueue.isEmpty()) {
					// 弹出一个最小权重的边(队列不为空)
					Edge edge = pQueue.poll();
					Node toNode = edge.to;
					if (!nodes.contains(toNode)) {
						nodes.add(toNode);
						edges.add(edge);
						//将toNode的边都解锁,加进优先级队列
						for(Edge nextEdge:toNode.edges){
							pQueue.add(nextEdge);
						}
					}
				}
			}
		}
		return edges;
	}
}