发展采矿业当然首先得有矿井,小 FF 花了上次探险获得的千分之一的财富请人在岛上挖了 nn 口矿井,但他似乎忘记了考虑矿井供电问题。
为了保证电力的供应,小 FF 想到了两种办法:
在矿井 ii 上建立一个发电站,费用为 vivi(发电站的输出功率可以供给任意多个矿井)。
将这口矿井 ii 与另外的已经有电力供应的矿井 jj 之间建立电网,费用为 pi,jpi,j。
小 FF 希望你帮他想出一个保证所有矿井电力供应的最小花费方案。
输入格式
第一行包含一个整数 nn,表示矿井总数。
接下来 nn 行,每行一个整数,第 ii 个数 vivi 表示在第 ii 口矿井上建立发电站的费用。
接下来为一个 n×nn×n 的矩阵 PP,其中 pi,jpi,j 表示在第 ii 口矿井和第 jj 口矿井之间建立电网的费用。
数据保证 pi,j=pj,ipi,j=pj,i,且 pi,i=0pi,i=0。
输出格式
输出一个整数,表示让所有矿井获得充足电能的最小花费。
数据范围
1≤n≤3001≤n≤300,
0≤vi,pi,j≤1050≤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
分析:对于这道题,我们可以假设一个虚拟的”超级发电站”.
而在每一个矿井修建发电站就相当于是从超级发电站向其连一条权值为v[i]的边.
之后我们惊奇的发现这就是一道最小生成树的模板题.
直接上克鲁斯卡尔算法即可..
时间复杂度 O(nlogn)
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 310, M = 100010;
int n,m;
struct Edge
{
int a, b, w;
bool operator< (const Edge &t)const
{
return w < t.w;
}
}e[M];
int p[N];
int find(int x)
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main()
{
cin >> n;
for (int i = 1; i <= n; i ++ ) p[i] = i;
for (int i = 1; i <= n; i ++ )
{
int a;
cin >> a;
e[m++] = {n+1, i, a};
}
for (int i = 1; i <= n; i ++ ){
for (int j = 1; j <= n; j ++ ){
int a;
cin >> a;
if(i>j)
e[m++] = {i, j, a};
}
}
sort(e, e + m);
int res = 0;
for (int i = 0; i < m; i ++ )
{
int a = find(e[i].a), b = find(e[i].b), w = e[i].w;
if (a != b) {
p[a] = b;
res += w;
}
}
cout << res << endl;
return 0;
}