POJ1258--贪心&最小生成树的prim算法

来源:互联网 发布:windows office 密钥 编辑:程序博客网 时间:2024/04/30 15:29

用prim算法实现,T1集合为所有点集合,T2集合表示已经是最小生成树中的点
lowcost[maxn]用来实现集合T1内各顶点到集合T2中各顶点的权值最小边的权值
nearvex[maxn]用来实现集合T1内顶点距集合T2中哪个顶点的距离最小

prim算法的思想:

初始:lowcost[k]=edge[v0][k] , nearvex[k]=v0;其中v0指的是从哪点来构造最小生成树。

当lowcost[i]=-1时我们让它表示i顶点加入到了T2集合,

什么样的情况下我们将i结点加入到T2集合呢?

我们选择那些在T1集合中,距离T2集合中点最小的lowcost[k],将该点k加入到集合T2中,

加入后我们应该改变那些在T1集合中点到T2集合中的最小权值,如果edge[k][i]<lowcost[i]

则将lowcost[i]=edge[k][i];nearvex[i]=k;这就完成了一次操作,进行n-1次后,就能构成一个最小生成树

#include <stdio.h>#define inf 100000#define maxn 401int n;int lowcost[maxn];int nearvex[maxn];int edge[maxn][maxn];int prim(int u0){ int sumweight = 0; int i, j; for(i = 1; i <= n; i++)  //顶点从1开始 {  lowcost[i] = edge[u0][i];  nearvex[i] = u0; } lowcost[u0] = 0; nearvex[u0] = -1;   //lowcost[]为-1时表示该点已经加入T2 for(i = 1; i < n; i++)  //将n-1个顶点加入到顶点集合T2 {  int min = inf;  int v = -1;  for(j = 1; j <= n; j++)  {   if(nearvex[j] != -1 && lowcost[j] < min)   {    min = lowcost[j];    v = j;   }  }  if(v != -1)  {   //printf("%d %d %d\n", nearvex[v], v, lowcost[v]);   nearvex[v] = -1;   //把v加入T2中   sumweight += lowcost[v];   for(j = 1; j <= n; j++)   {    //更新lowcost[]数组    if(nearvex[j] != -1 && edge[v][j] < lowcost[j])//j在T1集合中&&并且满足v到j的距离小于j到T2中顶点的最小值    {     lowcost[j] = edge[v][j];     nearvex[j] = v;    }   }  } } return sumweight;}int main(){ int i, j; int result; while(scanf("%d", &n) != EOF) {  for(i = 1; i <= n; i++)   for(j = 1; j <= n; j++)   {    scanf("%d", &edge[i][j]);   }  result = prim(1);  printf("%d\n", result); } return 0;}


 

 

原创粉丝点击