搜索 继续畅通工程 problem e

Problem Description

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

 

Input

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

当N为0时输入结束。

 

Output

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

 

Sample Input

31 2 1 01 3 2 02 3 4 031 2 1 01 3 2 02 3 4 131 2 1 01 3 2 12 3 4 10

 

Sample Output

310

 

题目分析：先判断每条路的状态，如果是修好的，就让它们的成本是0；如果没修，就让成本给出的成本。这样问题的处理就和1233一样了。

1. #include<stdio.h>  
2. #include<string.h>  
3. #include<algorithm>  
4. using namespace std;  
5. #define inf 0x3f3f3f3f  
6. int map[100][100];  
7. int s[100],vis[100];  
8. int n,m;  
9. int prim()  
10. {  
11.     int i,j,t,p,min,cnt,minpos;  
12.     int ans=0;  
13.     cnt=0;  
14.     vis[1]=1;  
15.     s[cnt++]=1;  
16.     while(cnt<n)  
17.     {  
18.         t=cnt;  
19.         min=inf;  
20.         for(i=0;i<t;i++)  
21.         {  
22.             p=s[i];  
23.             for(j=1;j<=n;j++)  
24.             {  
25.                 if(!vis[j]&&map[p][j]<min)  
26.                 {  
27.                     min=map[p][j];  
28.                     minpos=j;  
29.                 }  
30.             }  
31.         }  
32.         ans+=min;  
33.         s[cnt++]=minpos;  
34.         vis[minpos]=1;  
35.     }  
36.     return ans;  
37. }  
38. int main()  
39. {  
40.     int i,sum;  
41.     while(~scanf("%d",&n)&&n)  
42.     {  
43.         memset(vis,0,sizeof(vis));  
44.         memset(map,inf,sizeof(map));  
45.         int b,c,d,sta;  
46.         m=n*(n-1)/2;  
47.         for(i=0;i<m;i++)  
48.         {  
49.             scanf("%d%d%d%d",&b,&c,&d,&sta);  
50.             if(sta==0)  
51.                 map[b][c]=map[c][b]=d;  
52.             else  
53.                 map[b][c]=map[c][b]=0;  
54.         }  
55.         sum=prim();  
56.         printf("%d\n",sum);  
57.     }  
58.     return 0;  
59. }