您的位置:首页 > 编程语言 > Java开发

最小支撑(生成)树的prim算法 java

2017-06-25 15:39 399 查看
package prim;
import java.util.*;

public class Prim {
static int MAX = Integer.MAX_VALUE;

public static void main(String[] args) {
int[][] map = new int[][] {
{ 0, 10, MAX, MAX, MAX, 11, MAX, MAX, MAX },
{ 10, 0, 18, MAX, MAX, MAX, 16, MAX, 12 },
{ MAX, MAX, 0, 22, MAX, MAX, MAX, MAX, 8 },
{ MAX, MAX, 22, 0, 20, MAX, MAX, 16, 21 },
{ MAX, MAX, MAX, 20, 0, 26, MAX, 7, MAX },
{ 11, MAX, MAX, MAX, 26, 0, 17, MAX, MAX },
{ MAX, 16, MAX, MAX, MAX, 17, 0, 19, MAX },
{ MAX, MAX, MAX, 16, 7, MAX, 19, 0, MAX },
{ MAX, 12, 8, 21, MAX, MAX, MAX, MAX, 0 } };
prim(map, map.length);
}
public static void prim(int[][] graph, int n){

char[] nodeName = new char[]{'A','B','C','D','E','F','G','H','I'};
int[] lowCost = new int
;  //到新集合的最小权
int[] mid= new int
;//存取前驱结点
List<Character> list=new ArrayList<Character>();//用来存储加入结点的顺序
int i, j, min, minid , sum = 0;
//初始化辅助数组 选取A点为第一个节点
for(i=1;i<n;i++)
{
lowCost[i]=graph[0][i];
mid[i]=0;
}
list.add(nodeName[0]);
//一共需要加入n-1个点
for(i=1;i<n;i++)
{
min=MAX;
minid=0;
//每次找到距离集合最近的点
for(j=1;j<n;j++)
{
if(lowCost[j]!=0&&lowCost[j]<min)
{
min=lowCost[j];
minid=j;
}
}

if(minid==0) return;//孤立点返回
list.add(nodeName[minid]);//将最小权值的点加入路径中
lowCost[minid]=0;//已经经过的点权值归0 防止出现圈
sum+=min;
System.out.println(nodeName[mid[minid]] + "到" + nodeName[minid] + " 权值:" + min);
//加入该点后,更新其它点到集合的距离
for(j=1;j<n;j++)
{
if(lowCost[j]!=0&&lowCost[j]>graph[minid][j])//在新加入的点的权集中  与原有的权值进行取小   保证不成圈
{
lowCost[j]=graph[minid][j];
mid[j]=minid;
}
}
}
System.out.println("sum:" + sum);

}

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: