还是畅通工程(MST)

还是畅通工程

Time Limit:2000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Submit Status

Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。 

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。 
当N为0时，输入结束，该用例不被处理。 

 

Output

对每个测试用例，在1行里输出最小的公路总长度。 

 

Sample Input

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

 

Sample Output

3 5

Hint

Hint  Huge input, scanf is recommended.
        

 

#include<stdio.h>
#include<iostream>
#include<string.h>
#define maxn 0x3fff
using namespace std;
int d[110] , map[110][110] , v[110] ;//d[i]用来存放(i - 1) - th ~ i - th (point) 的权值 ， v[i] 中0表示没选过 ， 1表示选过
int n , m ;

int prim (int start )
{
    int mm , x ;
    for (int i = 1 ; i <= n ; i++)
        d[i] = i == start ? 0 : maxn ; //给 起始点 和 非起始点 涂色
    for (int i = 1 ; i <= n ; i++) {
        mm = maxn ;
        for (int j = 1 ; j <= n ; j++) { //找出权值最小的边
            if ( !v[j] && d[j] < mm ) {
                mm = d[j] ;
                x = j ;
            }
        }
        v[x] = 1 ;
        for (int j = 1 ; j <= n ; j++ )  {
             if ( !v[j] && d[j] > map[x][j] && map[x][j] != maxn )
                d[j] = map[x][j] ;
        }
    }
    int tmp = 0 ;
    for (int i = 1 ; i <= n ; i++)
        tmp += d[i] ;
    return tmp ;
}

int main ()
{
   // freopen ("a.txt" , "r" , stdin ) ;
    int val ;
    int a , b ;
    int res ;
    while ( cin >> n , n) {
        m = n * (n - 1) >> 1 ;
        memset (map , 0x3f , sizeof(map) ) ;
        memset (v , 0 , sizeof(v) ) ;
         for(int i = 1 ; i <= m ; i++ )
        {
            scanf ("%d%d%d", &a , &b , &val );
            if( val < map[a][b] )
            {
                map[a][b] = val;
                map[b][a] = val;
            }
        }
        res = prim(1) ;
        cout << res << endl ;
    }
    return 0 ;
}

#include<stdio.h>
#include<algorithm>
using namespace std;
int fa[110] , n , m ;
struct edge
{
    int u , v , w ;
}e[20000];

int cmp (edge a , edge b)
{
    return a.w < b.w ;
}

int find (int x)
{
    return fa[x] == x ? x : find(fa[x]) ;
}

void kruskal ()
{
    sort (e + 1 , e + m + 1 , cmp ) ;
    int ans = 0 , x , y ;
    for (int i = 1 ; i <= m ; i++ ) {
        x = find (e[i].u) ;
        y = find (e[i].v) ;
        if ( x != y) {
            ans += e[i].w ;
            fa[x] = y ;
        }
    }
    printf ("%d
" , ans ) ;
}
int main ()
{
   // freopen ("a.txt" , "r" , stdin ) ;
    while (~ scanf ("%d" , &n ) , n ) {
        for (int i = 1 ; i <= n ; i++)
            fa[i] = i ;
        m = n * (n - 1) >> 1 ;
        for (int i = 1 ; i <= m ; i++ ) {
            scanf ("%d%d%d" , &e[i].u , &e[i].v , &e[i].w ) ;
        }
        kruskal () ;
    }
    return 0 ;
}