直接做不太好做,我们搞一个超级源点,让它向每一个点建一条权值为w的边,然后我们求最小生成树即可。这样可以保证至少有一个点我们建了水库。而且所有点连通。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define pa pair<int,int>
#define N 310
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
int n,a[N][N],d[N],ans=0;
bool vis[N];
void Prim(){
priority_queue<pa,vector<pa>,greater<pa> >q;memset(d,0x3f,sizeof(d));
d[0]=0;q.push(make_pair(0,0));
while(!q.empty()){
int x=q.top().second;q.pop();if(vis[x]) continue;
vis[x]=1;ans+=d[x];
for(int i=1;i<=n;++i)
if(a[x][i]&&a[x][i]<d[i])
d[i]=a[x][i],q.push(make_pair(d[i],i));
}
}
int main(){
// freopen("a.in","r",stdin);
n=read();
for(int i=1;i<=n;++i) a[i][0]=a[0][i]=read();
for(int i=1;i<=n;++i)
for(int j=1;j<=n;++j) a[i][j]=read();
Prim();
printf("%d\n",ans);
return 0;
}