POJ -1258 Agri-Net

#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;
}