hihocoder 最小生成树一·Prim算法

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

最近，小Hi很喜欢玩的一款游戏模拟城市开放出了新Mod，在这个Mod中，玩家可以拥有不止一个城市了！

但是，问题也接踵而来——小Hi现在手上拥有N座城市，且已知这N座城市中任意两座城市之间建造道路所需要的费用，小Hi希望知道，最少花费多少就可以使得任意两座城市都可以通过所建造的道路互相到达（假设有A、B、C三座城市，只需要在AB之间和BC之间建造道路，那么AC之间也是可以通过这两条道路连通的）。

提示：不知道为什么Prim算法和Dijstra算法很像呢Σ(っ °Д °;)っ 。

输入

每个测试点（输入文件）有且仅有一组测试数据。

在一组测试数据中：

第1行为1个整数N，表示小Hi拥有的城市数量。

接下来的N行，为一个N*N的矩阵A，描述任意两座城市之间建造道路所需要的费用，其中第i行第j个数为Aij，表示第i座城市和第j座城市之间建造道路所需要的费用。

对于100%的数据，满足N<=10^3，对于任意i，满足Aii=0，对于任意i, j满足Aij=Aji, 0<Aij<10^4.

输出

对于每组测试数据，输出1个整数Ans，表示为了使任意两座城市都可以通过所建造的道路互相到达至少需要的建造费用。

代码：

import java.util.Scanner;

public class Main {
	public int prim(int[][] matrix,int start){
		int n=matrix.length;
		int[] dp=new int[n];
		for(int i=0;i<matrix[0].length;i++){
			dp[i]=matrix[start][i];
		}
		boolean[] flag=new boolean[n];
		flag[start]=true;
		for(int k=1;k<n;k++){
			int pos=-1;
			int min=Integer.MAX_VALUE;
			for(int i=0;i<dp.length;i++){
				if(!flag[i]&&dp[i]<min){
					pos=i;min=dp[i];
				}
			}
			if(pos==-1) return Integer.MAX_VALUE;
			flag[pos]=true;
			for(int i=0;i<dp.length;i++){
				if(!flag[i]) dp[i]=Math.min(dp[i],matrix[pos][i]);				
			}
		}
		int re=0;
		for(int i=0;i<dp.length;i++){
			re+=dp[i];
		}
		return re;
	}
	public static void main(String[] args) {
		Main main=new Main();
		Scanner scan=new Scanner(System.in);
		int num=scan.nextInt();
		int[][] matrix=new int[num][num];
		for(int i=0;i<num;i++){
			for(int j=0;j<num;j++){
				matrix[i][j]=scan.nextInt();
			}
		}
		System.out.println(main.prim(matrix,0));
	}
}