继续畅通工程

http://acm.hdu.edu.cn/showproblem.php?pid=1879

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
using namespace std;
int n,tt;
int tx[102],ty[102];
struct node
{
    int x,y;
    int w;
    int flag;
}edge[101*100/2+1];
int bin[102];
void add(int u,int v,int w1,int e)
{
    edge[tt].x=u;
    edge[tt].y=v;
    edge[tt].w=w1;
    edge[tt].flag=e;
    tt++;
}
int cmp(const void *a,const void *b)
{
    struct node *aa=(struct node *)a;
    struct node *bb=(struct node *)b;
    if(aa->flag!=bb->flag)
        return bb->flag-aa->flag;
    else return aa->w-bb->w;
}
int findx(int x)
{
    int r=x;
    while(r!=bin[r])
        r=bin[r];
    int j,k;
    j=x;
    while(j!=r)
    {
        k=bin[j];
        bin[j]=r;
        j=k;
    }
    return r;
}
void merge(int x,int y)
{
    int fx=findx(x);
    int fy=findx(y);
    if(fx!=fy)
        bin[fx]=fy;
}
void Kuscal()
{
    int sum=0;
    int ll=1;//最小生成树n-1条边
    int i=0;
    while(ll<n&&i<tt)
    {
        if(ll==n) break;
        if(findx(edge[i].x)!=findx(edge[i].y))
        {
            merge(findx(edge[i].x),findx(edge[i].y));
            if(edge[i].flag==0)
            sum=sum+edge[i].w;
            ll++;
        }
        i++;
    }
    printf("%d\n",sum);


}
int main()
{
    int x1,y1,z1,flag1;
    while(scanf("%d",&n)!=EOF&&n!=0)
    {
        tt=0;
        for(int i=1;i<=n;i++)
            bin[i]=i;
       for(int i=0;i<n*(n-1)/2;i++)
        {
            scanf("%d%d%d%d",&x1,&y1,&z1,&flag1);
            add(x1,y1,z1,flag1);
        }
        qsort(edge,tt,sizeof(edge[0]),cmp);
       Kuscal();
    }
    return 0;
}
posted @ 2014-06-26 11:04  人艰不拆_zmc  阅读(195)  评论(0编辑  收藏  举报