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