还是畅通工程(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[] , map[][] , v[] ;//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 =  ; i <= n ; i++)
         d[i] = i == start ?  : maxn ; //给 起始点 和 非起始点 涂色
     for (int i =  ; i <= n ; i++) {
         mm = maxn ;
         for (int j =  ; j <= n ; j++) { //找出权值最小的边
             if ( !v[j] && d[j] < mm ) {
                 mm = d[j] ;
                 x = j ;
             }
         }
         v[x] =  ;
         for (int j =  ; j <= n ; j++ )  {
              if ( !v[j] && d[j] > map[x][j] && map[x][j] != maxn )
                 d[j] = map[x][j] ;
         }
     }
     int tmp =  ;
     for (int i =  ; 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 - ) >>  ;
         memset (map , 0x3f , sizeof(map) ) ;
         memset (v ,  , sizeof(v) ) ;
          for(int i =  ; 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() ;
         cout << res << endl ;
     }
     return  ;
 }

200 ms prim

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

 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 +  , e + m +  , cmp ) ;
     int ans =  , x , y ;
     for (int i =  ; 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\n" , ans ) ;
 }
 int main ()
 {
    // freopen ("a.txt" , "r" , stdin ) ;
     while (~ scanf ("%d" , &n ) , n ) {
         for (int i =  ; i <= n ; i++)
             fa[i] = i ;
         m = n * (n - ) >>  ;
         for (int i =  ; i <= m ; i++ ) {
             scanf ("%d%d%d" , &e[i].u , &e[i].v , &e[i].w ) ;
         }
         kruskal () ;
     }
     return  ;
 }

kruskal