首页 > 代码库 > prime算法的使用
prime算法的使用
prime算法的使用
package PrimeApplication;import java.util.Scanner;/** * 农民要建立互联网络,目的使村庄里所有的农民连上网,并且总费用最小。 * 多组数据,每组数据给出一个n, * 然后给出n * n大小的无向图的邻接矩阵表示,值表示边权。 * 要求输出最小生成树的权值和。 * */public class Main { public static int MAXCOST = Integer.MAX_VALUE;//整数的最大值 public static void main(String[] args) { Scanner input = new Scanner(System.in); int cost ; while(input.hasNext()){ int n = input.nextInt(); int [][]adjMat = new int[n+1][n+1]; for(int i=1;i<=n;i++){ for(int j=1;j<=n;j++){ adjMat[i][j] = MAXCOST; } } for(int i=1;i<=n;i++){ for(int j=1;j<=n;j++){ adjMat[i][j] = input.nextInt(); } } cost = prime(adjMat,n); System.out.println(cost); } } private static int prime(int[][] graph, int n) { /* lowcost[i]记录以i为终点的边的最小权值,当lowcost[i]=0时表示终点i加入生成树 */ int lowcost[]=new int[n+1]; /* mst[i]记录对应lowcost[i]的起点,当mst[i]=0时表示起点i加入生成树 */ int mst[]=new int[n+1]; int min, minid, sum = 0; /* 默认选择1号节点加入生成树,从2号节点开始初始化 */ for (int i = 2; i <= n; i++){ /* 最短距离初始化为其他节点到1号节点的距离 */ lowcost[i] = graph[1][i]; /* 标记所有节点的起点皆为默认的1号节点 */ mst[i] = 1; } /* 标记1号节点加入生成树 */ mst[1] = 0; /* n个节点至少需要n-1条边构成最小生成树 */ for (int i = 2; i <= n; i++){ min = MAXCOST; minid = 0; /* 找满足条件的最小权值边的节点minid */ for (int j = 2; j <= n; j++){ /* 边权值较小且不在生成树中 */ if (lowcost[j] < min && lowcost[j] != 0){ min = lowcost[j]; minid = j; } } /* 输出生成树边的信息:起点,终点,权值 */ //System.out.printf("%c - %c : %d\n", mst[minid] + ‘A‘ - 1, minid + ‘A‘ - 1, min); /* 累加权值 */ sum += min; /* 标记节点minid加入生成树 */ lowcost[minid] = 0; /* 更新当前节点minid到其他节点的权值 */ for (int j = 2; j <= n; j++){ /* 发现更小的权值 */ if (graph[minid][j] < lowcost[j]){ /* 更新权值信息 */ lowcost[j] = graph[minid][j]; /* 更新最小权值边的起点 */ mst[j] = minid; } } } /* 返回最小权值和 */ return sum; }}
prime算法的使用
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。