1146. 新的开始

发展采矿业当然首先得有矿井，小 F 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井，但他似乎忘记了考虑矿井供电问题。

为了保证电力的供应，小 F 想到了两种办法：

1. 在矿井 i 上建立一个发电站，费用为 vi（发电站的输出功率可以供给任意多个矿井）。
2. 将这口矿井 i 与另外的已经有电力供应的矿井 j 之间建立电网，费用为 pi,j。

小 F 希望你帮他想出一个保证所有矿井电力供应的最小花费方案。

输入格式

第一行包含一个整数 n，表示矿井总数。

接下来 n 行，每行一个整数，第 i 个数 vi 表示在第 i 口矿井上建立发电站的费用。

接下来为一个 n×n 的矩阵 P，其中 pi,j 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用。

数据保证 pi,j=pj,i，且 pi,i=0。

输出格式

输出一个整数，表示让所有矿井获得充足电能的最小花费。

数据范围

1≤n≤300,
0≤vi,pi,j≤105

输入样例：

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

输出样例：

9

思路:

/*
为了供应电力，要么在当前位置 i 建发电站，要么与另外的已经有电力供应的矿井 j 之间建立电网

1、在当前位置i建发电站的费用是vi,建立虚拟结点S，相当于i号点到S号点的费用是vi

2、如图所示，求n个矿井电力供应的最小花费，等价于求n + 1个点的最小生成树
*/

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 310;
int g[N][N];
int dist[N];
int st[N];
int n;
int res;

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

        for (int j = 0; j < n + 1; j++)
        {
            if (!st[j] && dist[j] > g[t][j])
                dist[j] = g[t][j];
        }
    }
}

int main()
{
    cin >> n;
    memset(g, 0x3f, sizeof(g));

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

    prim();

    cout << res << endl;

    return 0;
}