POJ 1258 Agri-Net 最小生成树

题意：

　　 给出一个N。接下来有一个N * N的邻接矩阵A，第 i 行 第 j 列代表点  i   和  j 相距A[i][j].求连同所有点的最短路径，求最小生成树即可。

思路：

　　 用P算法，直接用邻接矩阵来存放数据就好了。

import java.util.Scanner;
public classMain {
    final static int MAXN = 100 + 3;
    final static int INF = 100000  + 3;
    static int[][] arv = new int[MAXN][MAXN];
    static int[] used = new int[MAXN];
    static int[] lowCose = new int[MAXN];
    public static void main(String[] args){
        Scannercin = new Scanner(System.in);
        int N;
        while(cin.hasNext()){
            N = cin.nextInt();
            for(int i = 1; i <= N; i++){
                for(int j = 1; j <= N; j++){
                    arv[i][j] = cin.nextInt();
                    if(arv[i][j] == 0){
                        arv[i][j] = INF;
                    }
                }
            }
            int ans = prim(1, N);
            System.out.println(ans);
        }
        cin.close();
    }
    public static int prim(int sta, int N){
        for(int i = 1; i <= N; i++){
            lowCose[i] = arv[sta][i];
            used[i] = 0;
        }
        used[sta] = 1;
        int cntWgt = 0;
        for(int i = 1; i <= N -1; i++){
            int v = -1;
            int mini = INF;
            for(int j = 1; j <= N; j++){
                if(used[j] == 0 && mini > lowCose[j]){
                    mini = lowCose[j];
                    v = j;
                }
            }
            if(v != -1){
                cntWgt += lowCose[v];
                used[v] = 1;
                for(int j = 1; j <= N; j++){
                    if(used[j] == 0 && lowCose[j] > arv[v][j]){
                        lowCose[j] = arv[v][j];
                    }
                }
            }
        }
        return cntWgt;
    }
}