在von学长的讲解,各种资料,算法导论神马的,都观摩了一遍,外加这个博客和这个终于理解了。。。。
Prim的算法开始的时候理解错了,以为一直找最短的,再找到下个点再找最短的,结果WA了多次。。。无奈书上的模版代码实在是看不懂,经过询问学长后,终于理解找最小边的时候,要从集合中找,而不是下个点。。。终于在理解算法之后YY了个O(n^3)的代码。。。发现书上说此算法的复杂度为O(n^2),而且书上的模版和我的很不一样。。。终于,在学长的指导下,明白书上的代码了。
算法实现:用low[]来存已经在树上的节点,到每个节点的最短的值,如果没有边,就初始化为正无穷,从low[]找最小边,然后改变low数组,看代码的时候死活看不懂。。。自己写的那个直接是用一个队列把树上的点直接存起来,历遍一遍,所以复杂度成了O(n^3)。。。low[]数组有点运用DP的思想。。。还有prim一直只是一棵树,不断加边。
O(n^3)
1 #include <stdio.h> 2 #include <stdlib.h> 3 #include <string.h> 4 int p[101][101],o[101],key[101],n; 5 int main() 6 { 7 int i,j,k,sum,num,min; 8 scanf("%d",&n); 9 memset(o,0,sizeof(o)); 10 for(i = 1; i <= n; i ++) 11 for(j = 1; j <= n; j ++) 12 { 13 scanf("%d",&p[i][j]); 14 } 15 sum = 0; 16 key[1] = 1; 17 o[1] = 1; 18 for(num = 2; num <= n; num ++)//共操作n-1次 19 { 20 min = 9999999; 21 for(i = 1; i <= num-1; i ++)//队列历遍,找最短边 22 { 23 for(j = 1; j <= n; j ++) 24 { 25 if(o[j] == 0&&p[key[i]][j] < min) 26 { 27 min = p[key[i]][j]; 28 k = j; 29 } 30 } 31 } 32 o[k] = 1; 33 key[num] = k; 34 sum += min; 35 } 36 printf("%d\n",sum); 37 return 0; 38 }
O(n^2)
1 #include <stdio.h> 2 #include <stdlib.h> 3 #include <string.h> 4 int p[101][101],o[101],low[101],n; 5 int main() 6 { 7 int i,j,k,sum,num,min; 8 scanf("%d",&n); 9 memset(o,0,sizeof(o)); 10 for(i = 1; i <= n; i ++) 11 for(j = 1; j <= n; j ++) 12 { 13 scanf("%d",&p[i][j]); 14 } 15 sum = 0; 16 low[1] = 1; 17 o[1] = 1; 18 for(i = 1;i <= n;i ++)//初始化low数组 19 { 20 low[i] = p[1][i]; 21 } 22 for(num = 2; num <= n; num ++) 23 { 24 min = 9999999; 25 for(i = 1; i <= n; i ++)//找到生成树最小边 26 { 27 if(o[i] == 0&&low[i] < min) 28 { 29 min = low[i]; 30 k = i; 31 } 32 } 33 o[k] = 1;//标记
34 for(i = 1;i <= n;i ++)改变low数组,如果存在到某点的距离更短,把low改变。 35 { 36 if(low[i] > p[k][i]) 37 low[i] = p[k][i]; 38 } 39 sum += min; 40 } 41 printf("%d\n",sum); 42 return 0; 43 }