每个点向井连边,边权为其点权。跑MST就行力。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
struct edge{
int u,v,w;
bool operator <(const edge &b)const{return w<b.w;}
}e[maxn];
int cnt=0;
int n;
int fa[305];
int g[305][305];
int Find(int x){
return x==fa[x]?x:fa[x]=Find(fa[x]);
}
int main(){
cin>>n;
for(int i=1;i<=n;i++){
int x;cin>>x;
e[++cnt].u=i,e[cnt].v=n+1,e[cnt].w=x;
}
for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)cin>>g[i][j];
for(int i=1;i<n;i++){
for(int j=i+1;j<=n;j++){
e[++cnt].u=i,e[cnt].v=j,e[cnt].w=g[i][j];
}
}
for(int i=1;i<=n+1;i++)fa[i]=i;
sort(e+1,e+cnt+1);
int W=0;
for(int i=1;i<=cnt;i++){
if(Find(e[i].u)==Find(e[i].v))continue;
fa[Find(e[i].u)]=Find(e[i].v);
W+=e[i].w;
}
cout<<W;
return 0;
}