图论模板之–最小生成树

#include <cstdio>
#include <algorithm>
using namespace std;
int n,m;
long long ans;
struct {
    int x;
    int y;
    int z;
}a[100005];
int s[100055];
int fa[100005];
int cnt=0;
bool cmp(const node &x,const node &y)
{
    return x.z<y.z;
}
int find(int x)
{
    if(fa[x]==x)
    {
        return x;
    }
    return fa[x]=find(fa[x]);
}
void kkk()
{
    int f1;
    int f2;
    int k=0;
    for(int i=1;i<=n;i++)
    fa[i]=i;
    for(int i=1;i<=cnt;i++)
    {
        f1=find(a[i].x);
        f2=find(a[i].y);
        if(f1!=f2)
        {
            ans=ans+a[i].z;
            fa[f1]=f2;
            k++;
            if(k==n-1)
            break;
        }
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&s[i]);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a[++cnt].z);
            a[cnt].x=i;
            a[cnt].y=j;
        }
        a[++cnt].x=i;
        a[cnt].y=n+1;
        a[cnt].z=s[i];
    }
    n++;
    sort(a+1,a+1+cnt,cmp);
    kkk();
    cout<<ans<<endl;
}