继续畅通工程

Description

省 政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表 中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

 

Input

测 试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示 未建。
当N为0时输入结束。

 

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

 

Sample Input

3 1 2 1 0 1 3 2 0 2 3 4 0 3 1 2 1 0 1 3 2 0 2 3 4 1 3 1 2 1 0 1 3 2 1 2 3 4 1 0

 

Sample Output

3 1 0

 

 

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
using namespace std;
struct node 
{
       int i, j, val, oo;
}T[5000];
bool cmp( node A, node B )
{
     return A.val < B.val;
}
int set[5000];
void make_set()
{
     for( int i = 0; i < 1000; i++ )
         set[i] = i;
}
int find( int x )
{
    return set[x] == x ? x: set[x]= find(set[x] );
} 
int merge( int x, int y )
{
    int x1 = find(x), y1 = find(y);
    if( x1 != y1 )
    {
          set[x1] = y1;
    } 
}
int kruskal( int count )
{
    sort( T, T+count, cmp );
    int result = 0;
    for( int i = 0; i < count; i++ )
    {
           if( find(T[i].i)!= find(T[i].j) )
           {
                  merge( T[i].i , T[i].j );
                    result += T[i].val;
           }
    }
    printf( "%d\n", result );
}
int main()
{
    int n;
    while( scanf( "%d", &n ), n )
    {
          make_set();
         int count = 0;    
         for( int i = 1; i <= n*(n-1)/2; i++ )
         {
              scanf( "%d%d%d%d", &T[i].i,&T[i].j, &T[i].val, &T[i].oo );
              count++;
              if( T[i].oo )
              {
                  int xx = find( T[i].i);
                  int yy = find( T[i].j );
                  if( xx != yy )
                     merge( xx, yy );
              }
         }
         kruskal( count );    
    }
}