题干:
发展采矿业当然首先得有矿井,小 F 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井,但他似乎忘记了考虑矿井供电问题。
为了保证电力的供应,小 F 想到了两种办法:
(1)在矿井 i 上建立一个发电站,费用为 vi(发电站的输出功率可以供给任意多个矿井)。
(2)将这口矿井 i 与另外的已经有电力供应的矿井 j 之间建立电网,费用为
p
i
,
j
p_{i,j}
pi,j。
小 F 希望你帮他想出一个保证所有矿井电力供应的最小花费方案。
输入格式
第一行包含一个整数 n,表示矿井总数。
接下来 n 行,每行一个整数,第 i 个数 vi 表示在第 i 口矿井上建立发电站的费用。
接下来为一个 n×n 的矩阵 P,其中
p
i
,
j
p_{i,j}
pi,j 表示在第 i口矿井和第 j口矿井之间建立电网的费用。
数据保证
p
i
,
j
=
p
j
,
i
p_{i,j}=p_{j,i}
pi,j=pj,i,且
p
i
,
i
p_{i,i}
pi,i=0。
输出格式
输出一个整数,表示让所有矿井获得充足电能的最小花费。
数据范围
1
≤
n
≤
300
1≤n≤300
1≤n≤300
0
≤
v
i
,
p
i
,
j
≤
1
0
5
0≤vi,p_{i,j}≤10^5
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
思路:
如果不考虑方法(1),则这就是个最小生成树模板题,但由于矿井必须连上发电厂才能供电,所以我们将发电场视为点0(设矿井从1到n编号),在i上建立发电站认为是从0到i建立一条边
然后简单证明下:首先只有一个发电站就是普通最小生成树,肯定正确;当存在多个发电站时,因为建了发电站的矿井已经通过0点连通了,其他点的连通问题就转换成了0~n这(n+1)个点的最小生成树。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int x[310][310],n,dis[310];
bool vis[310];
int prime(){
int ans=0;
memset(dis,0x3f,sizeof(dis));
dis[0]=0;
for(int i=0;i<=n;i++){
int s=-1;
for(int j=0;j<=n;j++){
if(!vis[j]&&(s==-1||dis[j]<dis[s]))
s=j;
}
vis[s]=true;
ans+=dis[s];
for(int j=0;j<=n;j++){
dis[j]=min(dis[j],x[s][j]);
}
}
return ans;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&x[0][i]);
x[i][0]=x[0][i];
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
scanf("%d",&x[i][j]);
}
}
printf("%d\n",prime());
return 0;
}