POJ -1258 Agri-Net

AC代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 200005
int p[N];
int n;
int L;
struct node
{
    int w,v,c;
}t[N];
int cmp(node p,node q)
{
    return p.c<q.c;
}
int find(int k)
{
    if(p[k]==k)
    {
        return p[k];
    }
    else
    {
        return find(p[k]);
    }
}
int cl()
{
    int sum=0,cnt=0;
    int i;
    for(i=1;i<=L;i++)
    {
        int x=find(t[i].w);
        int y=find(t[i].v);
        if(x!=y)
        {
            p[y]=x;
            cnt++;
            sum+=t[i].c;
        }
        if(cnt==n-1)
            return sum;
    }
    return sum;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        int i,j;
        for(i=0;i<=n*n;i++)
        {
            p[i]=i;
        }
        L=0;
        int r;
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++)
            {
                scanf("%d",&r);
                t[L].w=i;
                t[L].v=j;
                t[L].c=r;
                L++;
            }
        }
        sort(t,t+L,cmp);
        printf("%d\n",cl());
    }
    return 0;
}