hdu 1233 最小生成树

1. /***********************************************************************************************************************
2. map 存路径，值为权值； weight保存个点到源起点的权值； pre保存结点的前驱，即与源起点有路的下一个点
3. length 生成的最短距离 point 图上共有多少点 sign该点是否已经找过
5. 算法prim:
6. 从一个结点的子图开始构造生成树：选择连接当前子图和子图外结点的最小权边，将相应结点和边加入子图，直至将所有结点加入子图
7. ***********************************************************************************************************************/
9. #include <stdio.h>
10. #include <string>
12. #define SIZE 1100
13. #define INF 0x7fffffff
16. int map[1100][1100], weight[1100], pre[1100], length, point;
17. bool sign[1100];
19. void prim(int weight[],int map[][SIZE], int pre[], bool sign[], int &length, int point_num, int source)
20. { //source源起点，一定要是路径中有的
21. for(int i=1; i<=point_num; i++) //这里的路径的标号都 > 1，记录所以点到源起点的权值，前驱，将该点置为已经查找
22. {
23. weight[i] = map[source][i];
24. pre[i] = source; sign[i] = true;
25. }
26. sign[source] = false; length = 0;
29. for(int i=1; i<point_num; i++) //枚举n-1个点，即n-1个通路
30. {
31. int min = INF, sign_node = i; //sign_node 记录找到的最小的下一个点
32. for(int j=1; j<=point_num; j++) //查找最小权值的路径
33. {
34. if(sign[j] && weight[j] < min)
35. {min = weight[j]; sign_node = j;}
36. }
37. {sign[sign_node] = false; length += min; } //找到点置为已找到，长度相加,
38. //可以在这里添加一个标记，使他值等于min，如果最后值等于最大值，则不能生成最小生成树
40. for(int j=1; j<=point_num; j++) //重新设定源起点，将剩下的未找的点加入
41. {
42. if(weight[j] > map[sign_node][j] && sign[j] )
43. {weight[j] = map[sign_node][j]; pre[j] = sign_node;}
44. }
46. }
48. }
51. void input(int n, int map[][SIZE])
52. {
53. for(int i=1; i<=n ;i++)
54. {
55. int a, b, c;
56. scanf("%d%d%d", &a, &b, &c);
57. map[a][b] = c; map[b][a] = c;
58. }
59. }
62. int main()
63. {
64. int N;
65. while(~scanf("%d", &N) && N )
66. {
67. //scanf("%d", &M);
69. //int map[1100][1100], weight[1100], pre[1100], length, point;
70. //bool sign[1100];
72. for(int i=0; i<=N; i++) //初始化，N是端点个数
73. {
74. weight[i] = INF; pre[N] = 0;
75. for(int j=0; j<=N; j++)
76. map[i][j] = INF;
77. }
78. memset(sign, false, sizeof(sign) );
79. length = point = 0;
81. input( N*(N-1)/2, map);
82. prim(weight, map, pre, sign, length, N, 1); //查找，注意这个1 它是源起点，一定要是在路径上已知的点
84. printf("%d\n", length);
86. }
87. return 0;
88. }
90. /*
91. Problem Description
92. 某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
95. Input
96. 测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
97. 当N为0时，输入结束，该用例不被处理。
100. Output
101. 对每个测试用例，在1行里输出最小的公路总长度。
104. Sample Input
105. 3
106. 1 2 1
107. 1 3 2
108. 2 3 4
109. 4
110. 1 2 1
111. 1 3 4
112. 1 4 1
113. 2 3 3
114. 2 4 2
115. 3 4 5
116. 0
119. Sample Output
120. 3
121. 5
122. */

来自为知笔记(Wiz)

附件列表