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】