hud 1233 还是畅通工程_kruskal算法

#include <iostream>
#include <algorithm>

using namespace std;
#define MAXN 1005
struct node  //定义结构体
{
    int w,v,d;
};
node Edge[5000];
bool cmp(node a,node b) //结构体按照距离排序
{
    return a.d<b.d;
}
int n;
int Father[MAXN];
void Makeset()  //下面三个函数都属于并查集  用来判断是否会生成环
{
    for(int i=1; i<=n; i++)
        Father[i]=i;
}
int Find(int x)
{
    while(x!=Father[x])
        x=Father[x];
    return x;
}
void Combine(int x,int y)
{
    int a=Find(x);
    int b=Find(y);
    if(a!=b)
        Father[a]=b;
}
int main()
{
    int i,j,t,s,e,d;
    while(cin>>n,n)
    {
        int sum=0;
        Makeset();
        t=n*(n-1)/2;
        for(i=1; i<=t; i++)
        {
            cin>>s>>e>>d;
            Edge[i].w=s;
            Edge[i].v=e;
            Edge[i].d=d;
        }
        sort(Edge+1,Edge+t+1,cmp);
        for(j=1; j<=t; j++)
            if(Find(Edge[j].w)!=Find(Edge[j].v)) //每次增加一条边  判断是否形成环 
            {
                Combine(Edge[j].w,Edge[j].v);
                sum=sum+Edge[j].d;
            }
        cout<<sum<<endl;
    }
    return 0;
}




你可能感兴趣的:(ACM,图,HDU,kruskal)