【codevs 1078】最小生成树

#include <iostream>
#include <algorithm>
using namespace std;
struct data
{
int u,v,w;
}e[10001];
int father[101];
void init(int n)
{
for(int i=1;i<=n;i++)
father[i]=i;
}
int find(int x)
{
if(x!=father[x]) father[x]=find(father[x]);
return father[x];
}
void Union(int x,int y)
{
x=find(x);y=find(y);
if(x!=y) father[x]=y;
}
bool cmp(const data &a,const data &b)
{
return a.w<b.w;
}

int n,x,m,tot,k;
int main()
{
cin>>n;
init(n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
cin>>x;
if(x)
{
m++;
e[m].u=i;e[m].v=j;e[m].w=x;
}
}
sort(e,e+m+1,cmp);
for(int i=1;i<=m;i++)
{
if(find(e[i].u)!=find(e[i].v))
{
Union(e[i].u,e[i].v);
tot+=e[i].w;
k++;
}
if(k==n-1) break;
}
cout<<tot;
return 0;
}