最短网络【最小生成树】

最短网络

题目

思路：最小生成树模板题，直接prim

具体代码如下

#include<iostream>
#include<cstring>

using namespace std;

const int N = 110;

int n;
int g[N][N];
int dis[N];
bool st[N];

int prim(){
    memset(dis, 0x3f, sizeof dis);
    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 || dis[t] > dis[j]))
                t = j;
        st[t] = true;
        if(i && dis[t] == 0x3f3f3f3f) return 0x3f3f3f3f;
        if(i) res += dis[t];
        for(int j=1; j<=n; ++j) dis[j] = min(dis[j], g[t][j]);
    }
    return res;
}

int main(){
    
    scanf("%d", &n);
    for(int i=1; i<=n; ++i)
        for(int j=1; j<=n; ++j)
            scanf("%d", &g[i][j]);
            
    int t = prim();
    printf("%d\n", t);
    
    return 0;
}