关闭

最小生成树Prim

94人阅读 评论(0) 收藏 举报
分类:

优化算法

package mintree;
/*
 * prim算法的思想: 初始化时,v0加入到最小树,其他所有顶点作为未加入树的集合
 * 取矩阵中第一横,lowcost[],其实就是v0与其他顶点的距离,找出最小的,比如v4,v4加入到最小树,此时最小数有两个节点了v0和v4
 * 接下来,要找到其他未加入树顶点中与最小树顶点距离最近的那个点 lowcost[]这是v0的数据 找到v4与其他顶点的距离数据,即矩阵的第5横
 * tmp[] 然后rmp[]和lowcost[]纵向对比大小,小的数据设置到lowcost[]
 * 然后横向对比lowcost[]数据,找到最小点X,这个X即为与最小树距离最近的那个点 同理,依次将所有顶点加入到最小树
 */
public class Prim {


static int MAX = 65535;
public static void prim(int[][] graph, int n) {


char[] c = new char[] { 'A', 'B', 'C', 'D', 'E' };
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++) {//n-1次循环,也就是查找n-1个最小边
min = MAX;
minId = 0;
for (j = 1; j < n; j++) {
if (lowcost[j] < min && lowcost[j] != 0) {//横向比较lowcost[]
min = lowcost[j];
minId = j;
}
}
System.out.println(c[mst[minId]] + "到" + c[minId] + " 权值:" + min+"......."+i);


sum = sum + min;
lowcost[minId] = 0;


for (j = 1; j < n; j++) {
if (graph[minId][j] < lowcost[j]) {//对graph[minId][j]的理解
lowcost[j] = graph[minId][j];
mst[j] = minId;
}
}
}
System.out.println("sum:" + sum);
}

public static void main(String[] args) {
int[][] map = new int[][] {
{ 0, 4, 3, MAX, MAX },
{ 4, 0, 18, MAX, MAX }, 
{ 3, 18, 0, 22, 5 },
{ 0, MAX, 22, 0, 9 },
{ MAX, MAX, 5, 9, 0} };
prim(map, map.length);
}
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:4119次
    • 积分:171
    • 等级:
    • 排名:千里之外
    • 原创:12篇
    • 转载:15篇
    • 译文:0篇
    • 评论:0条