Prim算法

题目描述：

输入


第1行：2个数N,M中间用空格分隔，N为点的数量，M为边的数量。（2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行：每行3个数S E W，分别表示M条边的2个顶点及权值。(1 <= S, E <= N，1 <= W <= 10000)


输出


输出最小生成树的所有边的权值之和。

输入示例：

9 14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 9 7
2 8 11
3 9 2
7 9 6
3 6 4
4 6 14
1 8 8


输出示例


37

import java.util.Scanner;

/**
 * 第1行：2个数N,M中间用空格分隔，N为点的数量，M为边的数量。
 * （2 <= N <= 1000, 1 <= M <= 50000)
 * 第2 - M + 1行：每行3个数S E W，分别表示M条边的2个顶点及权值。
 * (1 <= S, E <= N，1 <= W <= 10000)
 * 输出最小生成树的所有边的权值之和
 */
public class Main{

    public static long prim(int[][] a, int N) {

        long res = 0;
        boolean[] vis = new boolean[N + 5];
        int[] min = new int[N + 5];
        init(min);
        min[1] = 0;
        for (int i = 1; i <= N; i++) {

            int k = -1;
            for (int j = 1; j <= N; j++) {
                if (!vis[j] && (k == -1 || min[j] < min[k])) {
                    k = j;
                }
            }
            vis[k] = true;
            res += min[k];
            for (int l = 1; l <= N; l++) {
                if (!vis[l] && a[k][l] < min[l])
                    min[l] = a[k][l];
            }
        }

        return res;
    }
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        int M = scanner.nextInt();
        int[][] a = new int[N + 5][N + 5];
        init(a);
        for (int i = 0; i < M; i++) {
            int s = scanner.nextInt();
            int e = scanner.nextInt();
            int w = scanner.nextInt();
            a[s][e] = a[e][s] = w;
        }
        System.out.println(prim(a, N));
    }

    public static void init(int[] a) {
        int n = a.length;
        for (int i = 0; i < n; i++) {
            a[i] = Integer.MAX_VALUE;
        }
    }
    public static void init(int[][] a) {
        int n = a.length;
        int m = a[0].length;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                a[i][j] = Integer.MAX_VALUE;
            }
        }
    }
}