题目大意:给定n个点,每个点可以花w[i]的代价建水井,或者花p[i][j]的代价连接到一个已经供水的点,求最小花销
将每个点向超级源连一条边,边权为w[i]
求最小生成树即可
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 310
using namespace std;
int n,map[M][M];
int Prim(int n)
{
static int f[M];
static bool v[M];
memset(f,0x3f,sizeof f);
f[1]=0;
int i,j,k,re=0;
for(j=1;j<=n;j++)
{
k=0;
for(i=1;i<=n;i++)
if(!v[i]&&f[i]<f[k])
k=i;
v[k]=1;re+=f[k];
for(i=1;i<=n;i++)
f[i]=min(f[i],map[k][i]);
}
return re;
}
int main()
{
int i,j,x;
cin>>n;
for(i=1;i<=n;i++)
{
scanf("%d",&x);
map[i][n+1]=map[n+1][i]=x;
}
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
scanf("%d",&map[i][j]);
cout<<Prim(n+1)<<endl;
return 0;
}