第十三周项目（1）——Prim算法的验证

1. /*      
2. * Copyright (c)2016,烟台大学计算机与控制工程学院      
3. * All rights reserved.      
4. * 文件名称：wu.cpp      
5. * 作    者：武昊      
6. * 完成日期：2016年12月1日      
7. * 版 本 号：v1.0       
8. *问题描述：验证Prim算法 
9. *输入描述：无      
10. *程序输出：测试数据      
11. */

程序中graph.h是图存储结构的“算法库”中的头文件：

代码：

1. #include <stdio.h>  
2. #include <malloc.h>  
3. #include "graph.h"  
4.   
5. void Prim(MGraph g,int v)  
6. {  
7.     int lowcost[MAXV];          //顶点i是否在U中  
8.     int min;  
9.     int closest[MAXV],i,j,k;  
10.     for (i=0; i<g.n; i++)           //给lowcost[]和closest[]置初值  
11.     {  
12.         lowcost[i]=g.edges[v][i];  
13.         closest[i]=v;  
14.     }  
15.     for (i=1; i<g.n; i++)           //找出n-1个顶点  
16.     {  
17.         min=INF;  
18.         for (j=0; j<g.n; j++)     //在(V-U)中找出离U最近的顶点k  
19.             if (lowcost[j]!=0 && lowcost[j]<min)  
20.             {  
21.                 min=lowcost[j];  
22.                 k=j;            //k记录最近顶点的编号  
23.             }  
24.         printf(" 边(%d,%d)权为:%d\n",closest[k],k,min);  
25.         lowcost[k]=0;           //标记k已经加入U  
26.         for (j=0; j<g.n; j++)       //修改数组lowcost和closest  
27.             if (g.edges[k][j]!=0 && g.edges[k][j]<lowcost[j])  
28.             {  
29.                 lowcost[j]=g.edges[k][j];  
30.                 closest[j]=k;  
31.             }  
32.     }  
33. }  
34.   
35. int main()  
36. {  
37.     MGraph g;  
38.     int A[6][6]=  
39.     {  
40.         {0,6,1,5,INF,INF},  
41.         {6,0,5,INF,3,INF},  
42.         {1,5,0,5,6,4},  
43.         {5,INF,5,0,INF,2},  
44.         {INF,3,6,INF,0,6},  
45.         {INF,INF,4,2,6,0}  
46.     };  
47.     ArrayToMat(A[0], 6, g);  
48.     printf("最小生成树构成:\n");  
49.     Prim(g,0);  
50.     return 0;  
51. }