POJ 1287 Networking （kuangbin带你飞 专题六：最小生成树）

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
int x,y,s;
};
int cmp(node a,node b)
{
return a.s<b.s;
}
int f[20000];
void inti()
{
for(int i=0;i<20000;i++)
f[i]=i;
}
int find(int x)
{
if(x!=f[x])
return f[x]=find(f[x]);
return x;
}
int main()
{
int n,m;
node s[30000];
while(cin>>n,n)
{
inti();
cin>>m;
int u,v,l;
for(int i=0;i<m;i++)
{
cin>>u>>v>>l;
s[i].x=u,s[i].y=v,s[i].s=l;
}
sort(s,s+m,cmp);
int ans=0,t=0;
for(int i=0;i<m;i++)
{
int x=find(s[i].x);
int y=find(s[i].y);
if(x!=y)
{
f[x]=y;
ans+=s[i].s;
t++;
}
if(t==n-1)
break;
}
cout<<ans<<endl;
}
return 0;
}