算法起步之Prim算法

原文:算法起步之Prim算法

prim算法是另一种最小生成树算法。他的安全边选择策略跟kruskal略微不同，这点我们可以通过一张图先来了解一下。

prim算法的安全边是从与当前生成树相连接的边中选择一条最短的一条，并且该边是应是生成树与生成树外一点的连接。

所以我们prim算法用汉字描述的过程应为：1初始化2构造最小优先队列，将所有节点都加入到最小优先队列中，所有节点的key设置为无穷大，开始节点设置成0。3循环，直到队列为空{取出key值最小的节点加入到生成树中，变量与key相连接的边，看是否是于树中的节点相连，如果不是且key值比队列中节点的key值小则更新队列中该节点的key值，最小优先队列排序}结束。

下面的代码就是按照刚才的过程实现的，有很多需要优化的地方，当然算法还需要根据具体的题目，选择最好的写法，这里只是给出思想。

public class Prim {

	private int MAX=1000;
	private int NULL=-1;

	public void prim(int[][]map){
		Node[] nodes=new Node[map.length];
		ArrayList list=new ArrayList();
		for (int i = 1; i < map.length; i++) {
			list.add(new Node(i, NULL, MAX));
		}

		list.add(new Node(0,NULL,0));
		while(!list.isEmpty()){
			Collections.sort(list);
			Node n=(Node) list.remove(1);
			System.out.println(n.getValue()+","+n.getLink()+","+n.getKey());
			nodes[n.getValue()]=n;
			for (int i = 0; i < map.length; i++) {
				if(map[n.getValue()][i]!=0&&nodes[i]==null){
					for (Object o : list) {
						Node node=(Node) o;
						if(node.getKey()==i&&map[n.getLink()][i]<node.getValue()){
							node.setLink(n.getValue());
							node.setKey(map[n.getLink()][i]);
						}
					}
				}

			}
		}
	}
}
class Node implements Comparable<Node>{
	private int value;
	private int link;
	private int key;
	public int getValue() {
		return value;
	}
	public void setValue(int value) {
		this.value = value;
	}
	public int getLink() {
		return link;
	}
	public void setLink(int link) {
		this.link = link;
	}
	public int getKey() {
		return key;
	}
	public void setKey(int key) {
		this.key = key;
	}
	public Node(int value, int link, int key) {
		this.value = value;
		this.link = link;
		this.key = key;
	}
	@Override
	public int compareTo(Node o) {
		if (this.key>o.getKey()) {
			return -1;
		}
		return 1;
	}

}

友情提示：转载请注明出处【作者idlear    博客http://blog.csdn.net/idlear/article/details/19577841】