并查集~最小生成树 HDU1233

并查集~最小生成树 HDU1233

为了学习最小生成树 这两天学习了并查集 做并查集的题碰到了一道最小生成树 自己搞定了这道题 非常的高兴哈~

#include < iostream >
#include < stdio.h >
#include < algorithm >
using   namespace  std;
int  father[ 101 ];

void  makeset( int  n)
{
    for(int i=1;i<=n;i++)
    {
         father[i]=i;
    }
}
int  findset( int  x)
{
    if(father[x]!=x)
    {
        father[x]=findset(father[x]);
    }
    return father[x];
}


int  Union( int  a, int  b)
{
    int x=findset(a);
    int y=findset(b);
    if(x==y)
    return 1;//如果两个点在同一颗树中 那么这两个点添加以后会形成回路 返回1
    else
    {
        father[x]=y;
     return 0;
     }
}

struct  country
{
    int first;
    int second;
    int distance;
} a[ 5001 ];

bool  cmp(country x,country y)
{
    return x.distance<y.distance;
}


int  main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
        break;
        makeset(n);
        int sum=0;
        for(int i=1;i<=n*(n-1)/2;i++)
        {
            scanf("%d %d %d",&a[i].first,&a[i].second,&a[i].distance);
        }
        sort(a+1,a+n*(n-1)/2+1,cmp);//将距离从小到大的排序
        for(int i=1;i<=n*(n-1)/2;i++)
        {
            if(Union(a[i].first,a[i].second)==0)//如果两个点在同一颗树中 那么这两个点添加以后会形成回路 否则这两个点之间的路加入到最小生成树中
              sum=sum+a[i].distance;
        }
        printf("%d\n",sum);
    }
     return 0;
}