1、基本思想
在prim算法中,通过加入最小邻接边的方法来建立最小生成树算法。首先构造一个零图,在选一个初始顶点加入到新集合中,然后分别在原先的顶点集合中抽取一个顶点,使得构成的边为权值最小,然后将该笔边加入到图中,并将抽出的顶点加入到新集合中,重复这个过程,知道新集合等于原先的集合。
2、基本代码
public class Prim {
static int MAX = 10000;
public static void main(String[] args) {
int[][] map = new int[][]{
{0,6,9,5,13},
{6,0,6,7,8},
{9,6,0,8,3},
{5,7,9,0,3},
{13,8,3,3,0}
};//定义边距
prim(map,map.length);
}
private static void prim(int[][] graph, int n) {
char[] c = new char[]{'1','2','3','4','5'};//定义每个点
int[] lowcost = new int[n];
int[] mst = new int [n];
int i,j,min,minid,sum=0;
for (i = 1; i < n; i++) {
lowcost[i] = graph[0][i];
mst[i] = 0;
}
for (i = 1; i < n; i++) {
min = MAX;
minid = 0;
for (j = 1; j < n; j++) {
if (lowcost[j]<min && lowcost[j]!=0) {
min = lowcost[j];
minid = j;
}
}//找到每一行的最小值
System.out.println(c[mst[minid]] + "到" + c[minid] + "权值" + min);
sum = sum + min;
for (j = 1; j < n; j++) {
if (graph[minid][j] < lowcost[j]) {
lowcost[j] = graph[minid][j];
mst[j] = minid;
}
}//找到下一个点的位置,再赋值
}
System.out.println(sum);
}
}
3、结果
1到4权值5
4到5权值3
5到3权值3
1到2权值6
17