acwing算法提高之图论--最小生成树的扩展应用

1 介绍

本专题用来记录使用最小生成树算法（prim或kruskal）解决的扩展题目。

2 训练

题目1：1146新的开始

C++代码如下，

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 310, INF = 0x3f3f3f3f3;
int n, m;
int g[N][N];
int d[N];
bool st[N];

void prim() {
   
    memset(d, 0x3f, sizeof d);
    int res = 0;
    
    for (int i = 0; i < n + 1; ++i) {
   
        int t = -1;
        for (int j = 0; j <= n; ++j) {
   
            if (!st[j] && (t == -1 || d[t] > d[j])) {
   
                t = j;
            }
        }
        
        st[t] = true;
        if (i) res += d[t];
        
        for (int j = 0; j <= n; ++j) {
   
            if (d[j] > g[t][j]) {
   
                d[j] = g[t][j];
            }
        }
    }
    
    cout << res << endl;
    
    return;
}

int main() {
   
    cin >> n;
    memset(g, 0, sizeof g);
    
    for (int i = 1; i <= n; ++i) {
   
        int t;
        cin >> t;
        g[0][i] = t;
        g[i][0] = t;
    }
    
    for (int i = 1; i <= n; ++i) {
   
        for (int j = 1; j <= n; ++j) {
   
            cin >> g[i][j];
        }
    }
    
    prim();

    return 0;   
}

题目2：1145北极通讯网络

C++代码如下，

#include <iostream>
#include <cstring>
#include <algorithm>
#include