1140 最短网络（prim算法）

1. 问题描述：

农夫约翰被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。约翰的农场的编号是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
来源：https://www.acwing.com/problem/content/description/1142/

2. 思路分析：

分析题目可以知道这是一道裸的prim题目，因为输入的是邻接矩阵所以可以使用prim算法求解，这样写起来会比较方便一点，默写prim算法的模板即可。

3. 代码如下：

from typing import List


class Solution:
    def prim(self, n: int, w: List[List[int]]):
        INF = 10 ** 10
        # dis记录连通块到点的最短距离
        dis = [INF] * (n + 1)
        # 起点是编号为1的点, 所以第一个点的距离应该设置为0
        dis[1] = 0
        # st列表标记哪些点已经在连通块中
        st = [0] * (n + 1)
        res = 0
        for i in range(1, n + 1):
            k = -1
            for j in range(1, n + 1):
                # 找到到连通块距离最小的点的编号
                if st[j] == 0 and (k == -1 or dis[k] > dis[j]):
                    k = j
            st[k] = 1
            res += dis[k]
            for j in range(1, n + 1):
                # 从当前找到的这个点开始更新到连通块的最短距离
                dis[j] = min(dis[j], w[k][j])
        return res

    def process(self):
        n = int(input())
        w = [[0] * (n + 1)]
        for i in range(1, n + 1):
            w.append([0] + list(map(int, input().split())))
        return self.prim(n, w)


if __name__ == "__main__":
    print(Solution().process())