HDOJ1879(继续畅通工程)

题目链接

最小生成树的题。克鲁斯卡尔算法。

#include <stdio.h>
#define N 100
#define M 5000
struct node
{
    int a,b,d;
}edge[M];
int n;
int p[N];
void make_set()
{
    int i;
    for(i=1;i<=n;i++)   p[i]=i;
}
int find_set(int i)
{
    return i==p[i]?p[i]:(p[i]=find_set(p[i]));
}
void union_set(int i,int j)
{
    i=find_set(i),j=find_set(j);
    p[j]=i;
}
int cmp(const void *a,const void *b)
{
    return ((struct node*)a)->d-((struct node*)b)->d;
}
int main()
{
    int i,ans,cnt,x,y,z,f;
    while(scanf("%d",&n)&&n)
    {
        make_set();
        cnt=0;
        for(i=0;i<n*(n-1)/2;i++)
        {
            scanf("%d%d%d%d",&x,&y,&z,&f);
            edge[i].a=x,edge[i].b=y,edge[i].d=z;
            if(f)   union_set(x,y),cnt++;
        }
        ans=0;
        qsort(edge,n*(n-1)/2,sizeof(edge[0]),cmp);
        for(i=0;cnt<n-1&&i<n*(n-1)/2;i++)
        {
            if(find_set(edge[i].a)==find_set(edge[i].b))    continue;
            ans+=edge[i].d;
            union_set(edge[i].a,edge[i].b);
            cnt++;
        }
        printf("%d\n",ans);
    }
    return 0;
}