SDUT 2048 最优布线问题(prim)

题目链接

在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 }

 

转载于:https://www.cnblogs.com/naix-x/archive/2012/06/14/2549630.html