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;
	}

}