使用超级源点的思想,先建立每个点到超级源点的的边,边权就是vi,这个边表示在这个矿井自建发电厂。接下来就是在这个n+1个点的图上,做一遍最小生成树算法。p表示通过连接的方式通电,v表示自己建立发电厂,在图上是选择连接到超级源点的边。
#include<iostream>
#include<cstring>
using namespace std;
const int N=310;
int g[N][N],dist[N];
int n;
bool st[N];
int prim(){
memset(dist,0x3f,sizeof dist);
int res=0;
dist[0]=0;
for(int i=0;i<n+1;i++){
int t=-1;
for(int j=0;j<=n;j++){
if(!st[j]&&(t==-1||dist[t]>dist[j]))
t=j;
}
st[t]=true;
res+=dist[t];
for(int j=0;j<=n;j++) dist[j]=min(dist[j],g[t][j]);
}
return res;
}
int main(){
cin>>n;
for(int i=1;i<=n;i++) {
scanf("%d",&g[i][0]);
g[0][i]=g[i][0];
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>g[i][j];
int t=prim();
cout<<t;
return 0;
}