Description
发展采矿业当然首先得有矿井,小 FF 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井,但他似乎忘记了考虑矿井供电问题。
为了保证电力的供应,小 FF 想到了两种办法:
在矿井 i 上建立一个发电站,费用为 vi(发电站的输出功率可以供给任意多个矿井)。
将这口矿井 i 与另外的已经有电力供应的矿井 j 之间建立电网,费用为 pi,j 。
小 FF 希望你帮他想出一个保证所有矿井电力供应的最小花费方案。
Input
第一行包含一个整数 n ( 1 ≤ n ≤ 300 ) ,表示矿井总数。
接下来 n 行,每行一个整数,第 i 个数 vi 表示在第 i 口矿井上建立发电站的费用。 0 ≤ vi ≤ 105
接下来为一个 n × n 的矩阵 P,其中 pi,j 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用。 0 ≤ pi,j ≤ 105
数据保证 pi,j = pj,i ,且 pi,i = 0
Output
一个整数,表示让所有矿井获得充足电能的最小花费
Sample Input
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0Sample Output
9
思路:建立一个超级源点(发电站)连接其它所有矿井,边的权重为在各矿井建立发电站的费用,求该图的最小生成树的权重即可
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define N 360
int g[N][N], dist[N], st[N];
int n;
int ans;
void prim() {
memset(dist, 0x3f, sizeof dist);
dist[0] = 0;
for (int i = 0; i <= n; i++) {
int t = -1;
for (int j = 0; j <= n; j++)
if (!st[j] && (t == -1 || dist[t] > dist[j]))
t = j;
st[t] = 1;
ans += dist[t];
for (int j = 0; j <= n; j++)
dist[j] = min(dist[j], g[t][j]);
}
}
int main() {
memset(g, 0x3f, sizeof g);
cin >> n;
int a;
for (int i = 1; i <= n; i++) {//令0为超级源点
cin >> a;
g[0][i] = g[i][0] = a;
}
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
cin >> g[i][j];
prim();
cout << ans <<endl;
return 0;
}
记录一种错误的思路:求原图的最小生成树权重加上所有矿井中建立发电站的最小费用即为答案
该思路错误的原因是只考虑建立一个发电站,但存在建立多个发电站的费用更低的情况,比如:每个矿井建立发电站的费用为1,但矿井之间连边的费用为100