AcWing 1140. 最短网络 最小生成树

AcWing 1140. 最短网络农夫

约翰被选为他们镇的镇长！

他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。

约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。

约翰的农场的编号是1，其他农场的编号是 2∼n。

为了使花费最少，他希望用于连接所有的农场的光纤总长度尽可能短。

你将得到一份各农场之间连接距离的列表，你必须找出能连接所有农场并使所用光纤最短的方案。

输入格式
第一行包含一个整数 n，表示农场个数。

接下来 n 行，每行包含 n 个整数，输入一个对角线上全是0的对称矩阵。
其中第 x+1 行 y 列的整数表示连接农场 x 和农场 y 所需要的光纤长度。

输出格式
输出一个整数，表示所需的最小光纤长度。

数据范围
3≤n≤100
每两个农场间的距离均是非负整数且不超过100000。

输入样例：

4
0  4  9  21
4  0  8  17
9  8  0  16
21 17 16  0

输出样式：

28

思路
典型的最小生成树问题，这里我们采用prim算法。

import java.util.*;
public class Main{
    static int n;
    static int [][]w=new int[110][110];
    static int []dist=new int[110];
    static boolean[]st=new boolean[110];
    public static void main(String[]args)
    {
        Scanner sc=new Scanner(System.in);
        n=sc.nextInt();
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                w[i][j]=sc.nextInt();
            }
        }
        System.out.println(prim());
    }
    static int prim()
    {
        Arrays.fill(dist,0x3f3f3f3f);
        dist[1]=0;
        int res=0;
        for(int i=0;i<n;i++)
        {
            int t=-1;
            for(int j=1;j<=n;j++)
            {
                if(!st[j]&&(t==-1||dist[t]>dist[j]))
                {
                    t=j;
                }
            }
            res+=dist[t];
            st[t]=true;
            //dist[j] 通过min(dist[j], w[t][j])维护连通块到达每个点的最小距离
        // 假设加进连通块的顺序为 1→2→3
        //t为连通块现在刚加进来的最小点 
        // 在只加进1的时候 dist[j]为所有点到1的距离 dist[j] = w[1][j]
        // t更新为2的时候 dist[j]= min(dist[j], w[t][j])
        //                 则更新为min(w[1][j],w[2][j]);
        // t更新为3的时候 dist[j]= min(dist[j], w[t][j])
        //                 则更新为min(w[1][j],w[2][j],w[3][j]);
            for(int j=1;j<=n;j++) dist[j]=Math.min(dist[j],w[t][j]);
        }
        return res;
    }
}