首页 > 代码库 > MST-kruskal

MST-kruskal

hdu 1233

#include<stdio.h>
#include<algorithm>
using namespace std;
struct dis{
    int a,b,c;
}s[10010];
int cmp(dis x,dis y)
{
    return x.c<y.c;
}
int z[110];
int f(int y)
{
    int r=y;
    while(r!=z[r])
    r=z[r];
    return r;
}
int merge(int a,int b)
{
    int fx=f(a);
    int fy=f(b);
    if(fx!=fy)
    {
        if(fx<fy)
        z[fy]=fx;
        else 
        z[fx]=fy;
        return 1;
    }
    else
    return 0;
}
int main()
{
    int t,i,n,sum,m;
    while(~scanf("%d",&t),t)
    {
        n=t*(t-1)/2;
        for(i=1;i<=t;i++)
        z[i]=i;
        for(i=0;i<n;i++)
        scanf("%d%d%d",&s[i].a,&s[i].b,&s[i].c);
        sort(s,s+n,cmp);
        m=1,sum=0;
        for(i=0;i<n&&m<t;i++)
        {
            if(merge(s[i].a,s[i].b))
            {
                m++;
                sum+=s[i].c;
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}

 

MST-kruskal