杭电 1879 继续畅通工程

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int set[110],n,m;
struct val
{
    int x,y,v,s;
}e[10000];
int cmp(const void *a,const void *b)
{
    return ((val *)a)->v-((val *)b)->v;
}
int find(int x)
{
    return x==set[x] ? x : find(set[x]);
}
int kruskal()
{
    int sum=0;
    int x,y,v,s;
    for(int i=1;i<=m;i++)
    {
        x=e[i].x,y=e[i].y,v=e[i].v,s=e[i].s;
        if(s==1)
        {
            int a=find(x),b=find(y);
            if(a!=b)
            {
               set[a]=b;
            }
        }
    }
    for(int i=1;i<=m;i++)
    {
        x=e[i].x,y=e[i].y,v=e[i].v,s=e[i].s;
        if(s==0)
        {
            int a=find(x),b=find(y);
            if(a!=b)
            {
                set[a]=b;
                sum+=v;
            }
        }
    }
    return sum;    
}
int main()
{
    int x,y,v,s;
    while(scanf("%d",&n),n)
    {
        for(int i=1;i<=n;i++)
        {
            set[i]=i;        
        }
        m=(n*(n-1))/2;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d%d",&x,&y,&v,&s);
            e[i].x=x,e[i].y=y,e[i].v=v,e[i].s=s;
        }
        qsort(e+1,m,sizeof(e[0]),cmp);
        int res=kruskal();
        printf("%d\n",res);
    }
   // system("pause");
    return 0;
}

转载于:https://www.cnblogs.com/zhangshu/archive/2011/07/19/2110452.html