hdu 1233 还是畅通工程 Kruskal 最小生成树 并查集

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;

struct edge
{
int u;
int v;
int w;

};
int p[100];
edge e[5000];

bool cmp(edge a,edge b)
{
return a.w<b.w;

}

int find(int x)
{
if(p[x]==x)  return x;
else
{
return p[x]=find(p[x]);
}
}
int n,m;

long long  kruskal()
{
for(int i=0;i<n;i++)
p[i]=i;

long long  ans=0;
for(int i=0;i<m;i++)
{
int x=find(e[i].u);
int y=find(e[i].v);
if(x!=y)
{
ans+=e[i].w;
p[x]=y;
}
}

return ans;
}
int main()
{

while(cin>>n)
{
if(n==0)   break;
m=n*(n-1)/2;
int a,b,c;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
e[i].u=a-1;
e[i].v=b-1;
e[i].w=c;
}
sort(e,e+m,cmp);

cout<<kruskal()<<endl;

}
}