POJ-1258:最小生成树裸题

题目链接
AC代码：（Kruskal实现）

#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
using namespace std;
typedef long long ll;
const ll N=105;
ll ans,n,fa[N];
struct Edge
{
    ll u,v,w;
}e[N*N];
bool comp(Edge x,Edge y)
{
    return x.w<y.w;//从小到大
}
void it(ll n)
{
    for(ll i=1;i<=n;i++)
        fa[i]=i;
}
ll fd(ll x)
{
    if(fa[x]!=x)
    {
        fa[x]=fd(fa[x]);
    }
    return fa[x];
}
ll mg(ll a,ll b)
{
    ll p=fd(a);
    ll q=fd(b);
    if(p==q) return 0;
    if(p>q) fa[p]=q;
    else fa[q]=p;
    return 1;
}
ll kls(ll n)
{
   ans=0;
   ll m=n*(n-1)/2;//边数
   for(ll i=0;i<m;i++)
        if(mg(e[i].u,e[i].v))
       {
           ans+=e[i].w;
           n--;
           if(n==1) return ans;
       }
       return 0;
}
int main()
{
 while(cin>>n)
 {
     memset(fa,0,sizeof(0));
     ll p=0;
     for(ll i=1;i<=n;i++)
        for(ll j=1;j<=n;j++)
        {
            ll w;
            cin>>w;
            if(i<j)
            {
                e[p].u=i;
                e[p].v=j;
                e[p++].w=w;
            }
        }
     it(n);
     sort(e,e+p,comp);
     ans=kls(n);
     cout<<ans<<endl;
 }
}