hdu 1233:还是畅通工程(数据结构,图,最小生成树,普里姆(Prim)算法)
还是畅通工程
Time Limit : 4000/2000ms (Java/Other) Memory Limit : 65536/32768K (Java/Other)
Total Submission(s) : 5 Accepted Submission(s) : 3
Font: Times New Roman | Verdana | Georgia
Font Size: ← →
Problem Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
当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.
Huge input, scanf is recommended.
Source
浙大计算机研究生复试上机考试-2006年
数据结构:图,最小生成树。
求最小生成树有两种算法可用,一种是普里姆算法,一种是克鲁斯卡尔(kruskal)算法。
这里我只使用了普里姆算法,并且直接套用的模板。
查找资料的时候发现一句话很经典,道出了两种算法分别适应的情况:
“克鲁斯卡尔(Kruskal)算法因为只与边相关,则适合求稀疏图的最小生成树。而prime算法因为只与顶点有关,所以适合求稠密图的最小生成树。”
不明白普里姆算法的可以见维基百科,上面讲解的很直白:维基百科 - 普里姆算法。
另外注意这道题输入尽量用C的模式,即使用scanf()函数,用C++的cin会超时。
本题代码:
1 #include <iostream>
2 #include <stdio.h>
3 #include <string.h>
4 using namespace std;
5 #define MAXN 110
6 #define UPPERDIS 9999999
7 int cost[MAXN][MAXN];
8 int n;
9 //普里姆算法模板
10 int lowcost[MAXN],closest[MAXN];
11 int prim(int v0)
12 {
13 int i,j,mindis,minone;
14 int ans = 0;/*用来记录最小生成树的总长度*/
15 /*各点距离初始化*/
16 for(i = 0;i < n;i++)
17 {
18 lowcost[i] = cost[v0][i];
19 closest[i] = v0;
20 }
21 for(i = 0;i < n-1;i++)
22 {
23 mindis = UPPERDIS;
24 for(j = 0;j < n;j++)
25 if(lowcost[j] && mindis > lowcost[j])
26 {
27 mindis = lowcost[j];
28 minone = j;
29 }
30 /*将找到的最近点加入最小生成树*/
31 ans += lowcost[minone];
32 lowcost[minone] = 0;
33 /*修正其他点到最小生成树的距离*/
34 for(j = 0;j < n;j++)
35 if(cost[j][minone] < lowcost[j])
36 {
37 lowcost[j] = cost[j][minone];
38 closest[j] = minone;
39 }
40 }
41 return ans;
42 }
43 int main()
44 {
45 while(scanf("%d",&n)!=EOF){
46 if(n==0) break;
47 memset(cost,0,sizeof(cost));
48 for(int i=0;i<n*(n-1)/2;i++){
49 int a,b,c;
50 scanf("%d%d%d",&a,&b,&c);
51 cost[a-1][b-1] = c;
52 cost[b-1][a-1] = c;
53 }
54 printf("%d\n",prim(0));
55 }
56 return 0;
57 }
Freecode : www.cnblogs.com/yym2013