继续畅通工程_MST(hdu 1879)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1879

Problem Description

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

Input

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

Output

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

题目分析：

这个题目其实简单的考查最小生成树MST ，关键在于对于状态的处理 ，如果该道路是已修通状态1 ,这这两点的距离设置为0。

代码如下：

View Code

 1 #include<cstdio>
 2 #include<iostream>
 3 using namespace std ;
 4 #define N 100 
 5 int dist[N][N] ; 
 6 int visit[N] ; 
 7 int lowCost[N] ;
 8 
 9 int main(void)
10 {
11     int n ; 
12     while(scanf("%d" , &n) != EOF && n != 0)
13     {
14         int i , j , x , y , value , flag , ans = 0 , min , iPos ;
15         for(i = 1 ; i <= n ; ++i)
16             for(j = 1 ; j <= n ; ++j)
17                 dist[i][j] = INT_MAX ;
18         for(i = 1 ; i <= n*(n-1)/2 ; ++i)
19         {
20             scanf("%d%d%d%d" , &x , &y , &value , &flag) ;
21             if(flag)
22                 dist[x][y] = dist[y][x] = 0 ;
23             else
24                 dist[x][y] = dist[y][x] = value ;
25         }
26         for(i = 1 ; i <= n ; ++i)
27         {
28             visit[i] = 0 ; 
29             lowCost[i] = dist[i][1] ;
30         }
31         visit[1] = 1 ; 
32         for(i = 1 ; i <= n ; ++i)
33         {
34             min = INT_MAX ; 
35             iPos = 1 ; 
36             for(j = 2 ; j <= n ; ++j)
37             {
38                 if(!visit[j] && min > lowCost[j])
39                 {
40                     min = lowCost[j] ; 
41                     iPos = j ;
42                 }
43             }
44             if(iPos == 1)
45                 break ; 
46             visit[iPos] = 1 ; 
47             ans += min ;
48             for(j = 2 ; j <= n ; ++j)
49             {
50                 if(!visit[j] && dist[j][iPos] < lowCost[j])
51                     lowCost[j] = dist[j][iPos] ;
52             }
53         }
54         printf("%d\n" , ans) ;
55     }
56     return 0 ;
57 }

 

 

 

 

转载于:https://www.cnblogs.com/CqlLiliang/archive/2012/07/05/2578330.html