prime算法详解



1. #include<iostream>  
2. #include<cstdio>  
3. #include<cstring>  
4. #include<string>  
5. #include<algorithm>  
6. using namespace std;  
7. #define INF 0x3f3f3f3f  
8. #define MAXN 110  
9. int map[MAXN][MAXN], lowcost[MAXN];  
10. bool visit[MAXN];  
11. int nodenum, sum;  
12.   
13. void prim()  
14. {  
15.     int temp, k;  
16.     sum = 0;  
17.     memset(visit, false, sizeof(visit)); //初始化visit  
18.     visit[1] = true;  
19.     for(int i = 1; i <= nodenum; ++i) //初始化lowcost[i]  
20.         lowcost[i] = map[1][i];  
21.     for(int i = 1; i <= nodenum; ++i)//找生成树集合点集相连最小权值的边  
22.     {  
23.         temp = INF;  
24.         for(int j = 1; j <= nodenum; ++j)  
25.             if(!visit[j] && temp > lowcost[j])  
26.                 temp = lowcost[k = j];  
27.         if(temp == INF) break;  
28.         visit[k] = true; //加入最小生成树集合  
29.         sum += temp;//记录权值之和  
30.         for(int j = 1; j <= nodenum; ++j) //更新lowcost数组  
31.             if(!visit[j] && lowcost[j] > map[k][j])  
32.                 lowcost[j] = map[k][j];  
33.     }  
34. }  
35.   
36. int main()  
37. {  
38.     int a, b, cost, edgenum;  
39.     while(scanf("%d", &nodenum) && nodenum)  
40.     {  
41.         memset(map, INF, sizeof(map));  
42.         edgenum = nodenum * (nodenum - 1) / 2;  
43.         for(int i = 1; i <= edgenum; ++i) //输入边的信息  
44.         {  
45.             scanf("%d%d%d", &a, &b, &cost);  
46.             if(cost < map[a][b])  
47.                 map[a][b] = map[b][a] = cost;  
48.         }  
49.         prim();  
50.         printf("%d\n", sum); //最小生成树权值之和  
51.     }  
52.     return 0;  
53. }