挺有意思的思路
如果不能自己打井,那么就是MST裸题了,考虑转换一下,自己打井就相当于连接一口虚拟的井(地下水?),所有井i到这口井的距离是w[i],这样把所有边排个序跑MST即可
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=305;
int n,a[N][N],cnt,f[N],con,ans;
struct qwe
{
int u,v,w;
}e[N*N];
bool cmp(const qwe &x,const qwe &y)
{
return x.w<y.w;
}
int read()
{
int r=0,f=1;
char p=getchar();
while(p>'9'||p<'0')
{
if(p=='-')
f=-1;
p=getchar();
}
while(p>='0'&&p<='9')
{
r=r*10+p-48;
p=getchar();
}
return r*f;
}
int zhao(int x)
{
return f[x]==x?x:f[x]=zhao(f[x]);
}
int main()
{
n=read();
for(int i=1;i<=n;i++)
a[i][n+1]=a[n+1][i]=read(),f[i]=i;
a[n+1][n+1]=1e9;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
a[i][j]=read();
for(int i=1;i<=n+1;i++)
for(int j=1;j<i;j++)
e[++cnt]=(qwe){i,j,a[i][j]};
sort(e+1,e+1+cnt,cmp);
for(int i=1;i<=cnt&&con<n;i++)
{
int fx=zhao(e[i].u),fy=zhao(e[i].v);
if(fx!=fy)
{
f[fx]=fy;
con++;
ans+=e[i].w;
}
}
printf("%d
",ans);
return 0;
}