洛谷 P1546：最短网络 ← Prim算法模板题

【题目来源】
https://www.luogu.com.cn/problem/P1546
https://www.acwing.com/problem/content/1142/

【题目描述】
农夫约翰被选为他们镇的镇长！
他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。
约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。
约翰的农场的编号是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 算法模板题。
另一道 Prim 算法模板题详见：https://blog.csdn.net/hnjzsyjyj/article/details/139605698

【算法代码】

#include <bits/stdc++.h>
using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=105;
int g[maxn][maxn];
int dis[maxn]; //dis存的是当前连通块与其外直接相连的点构成的所有边的长度的最小值
bool st[maxn];
int n;
int ans;

void prim() {
    memset(dis,inf,sizeof dis);
    for(int i=0; i<n; i++) {
        int t=-1; //最小边所连点的序号，初始化为-1
        for(int j=1; j<=n; j++) {
            if(!st[j] && (t==-1||dis[t]>dis[j]))
                t=j;
        }
        st[t]=true;
        if(i) ans+=dis[t];
        for(int j=1; j<=n; j++) {
            if(!st[j]) dis[j]=min(dis[j],g[t][j]);
        }
    }
    cout<<ans<<endl;
}

int main() {
    cin>>n;
    for(int i=1; i<=n; i++) {
        for(int j=1; j<=n; j++) {
            cin>>g[i][j];
        }
    }

    prim();

    return 0;
}

/*
in:
4
0  4  9  21
4  0  8  17
9  8  0  16
21 17 16  0

out:
28
*/

【参考文献】
https://blog.csdn.net/hnjzsyjyj/article/details/139605698
https://blog.csdn.net/hnjzsyjyj/article/details/139606429