[USACO3.1]最短网络 Agri-Net kruskal

题意

给出一个n个点邻接矩阵，问怎样花费最少的构造一个连通图。

 

版子

#include<iostream>#include<algorithm>#include<cstring>#include<cstdio>using namespace std;struct edge{int x,y,z;}a[101000];int fa[101000];int n,tot=0,ans;bool cmp(edge p,edge q){return p.z<q.z;}int getfa(int x){return fa[x]==x?fa[x]:fa[x]=getfa(fa[x]);}void kruskal(){    int cnt=0;    sort(a,a+tot,cmp);    for (int i=0;i<tot && cnt<n-1;i++)    {        int pp=getfa(a[i].x),qq=getfa(a[i].y);        if (pp!=qq)        {            fa[pp]=qq;            ans+=a[i].z;            cnt++;        }    }}int main(){    cin>>n;    for (int i=1;i<=n;i++) fa[i]=i;    for (int i=1;i<=n;i++)        for (int j=1;j<=n;j++)        {            int cur;            cin>>cur;            if (!cur) continue;            a[tot].x=i;a[tot].y=j;a[tot++].z=cur;        }    kruskal();    cout<<ans<<endl;    return 0;}