[P1550 [USACO08OCT]Watering Hole G]

P1550 [USACO08OCT]Watering Hole G

//可能是要选几个点呀
//增加一个超级起源点，把那些零散的树进行相互连接
#include <bits/stdc++.h>
using namespace std;
const int M=305;
 
struct Edge {
    int u,v,w;
    bool operator<(const Edge T)const {
        return w<T.w;
    }
}a[M*M];
int tot;
 
int fa[M];
int find(int x) {
    return x==fa[x]?fa[x]:fa[x]=find(fa[x]);
}
 
int main() {
    int n;cin>>n;
    for(int i=1;i<=n+1;i++)fa[i]=i;
    for(int i=1;i<=n;i++) {
        int x;cin>>x;
        a[++tot]={n+1,i,x};
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) {
            int x;cin>>x;
            a[++tot]={i,j,x};
        }
    sort(a+1,a+1+tot);
    int sum=0;
    for(int i=1;i<=tot;i++) {
        int fx=find(a[i].u),fy=find(a[i].v);
        if(fx!=fy)fa[fx]=fy,sum+=a[i].w;
    }
    cout<<sum;
    return 0;
    
}