题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=1233
还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 40310 Accepted Submission(s):
18325
Problem 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.
题意:给n个点,这些点之间有N(N-1)/2条路,每条路连接俩点,且有一定的距离(权),求把所有点连起来的最短路径
思路:采用prim方法把路从1号连到n号
1 #include<cstdio> 2 #include<string.h> 3 #include<algorithm> 4 using namespace std; 5 #define inf 0x3f3f3f3f 6 int map[105][105]; 7 int prim(int n) 8 { 9 int vis[105], dis[105]; 10 int ans = 0; 11 memset(vis, 0, sizeof(vis));//记录该点是否被访问过 12 memset(dis, inf, sizeof(dis)); 13 dis[1] = 0;//从第一个点开始至第n个点 14 for(int i = 1; i <= n; i++) 15 { 16 int k = -1, tp = inf; 17 for(int j = 1; j <= n; j++) 18 //寻找权最小的路 19 if(!vis[j] && dis[j] < tp) 20 { 21 k = j;//最小权连通的点(还未标记的) 22 tp = dis[j];//最小的权的大小(记为x) 23 } 24 vis[k] = 1;//标记新的点 25 ans += tp;//加入新的权值 26 for(int j = 1; j <= n; j++) 27 { //寻找新的最小权值 (仅次于x小的权值) 28 if(!vis[j] && map[k][j] < dis[j]) 29 dis[j] = map[k][j]; 30 } 31 } 32 return ans; 33 } 34 int main() 35 { 36 int n, m, a, b, c; 37 while( scanf("%d", &n), n) 38 { //初始化地图上的权,设为最大值 39 memset(map, inf, sizeof(map)); 40 m = n * (n - 1) / 2; 41 for(int i = 0; i < m; i++){ 42 scanf("%d %d %d", &a, &b, &c); 43 map[a][b] = map[b][a] = c;//无向图 44 } 45 printf("%d\n", prim(n)); 46 } 47 return 0; 48 }
时间: 2024-10-06 08:42:49