普里姆Prim— 最小生成树

 

#include<stdio.h>// 用prim算法构造五项连通图网的最小生成树
#define maxcost 1000
int gm[50][50];

void prim(int tree[],int cost[],int n)
{//从序号为0的顶点出发，建立连通网的最小生成树，二维数组gm[][]是其邻接矩阵
 //顶点编号依次为0......n-1,建立的最小生成树存于数组tree中，对应的边值在cost中

 int flag[50]={0};//标记数组，用于记录点是否加入到U中
 int i,j,k,mincost;

 for(i=0;i<n;i++)
 {
  cost[i]=gm[0][i];//从存储序号为0的顶点出发生成最小生成树
  tree[i]=0;
 }
// tree[0]=-1;

 flag[0]=1;//顶点0进入U

    for(i=1;i<n;i++)
 {
  mincost=maxcost;
  for(j=1;j<n;j++)
  {
   if(flag[j]==0&&cost[j]<mincost)
   {
     mincost=cost[j];
     k=j;//记忆最小的边
   }
  }
  
  flag[k]=1;//k加入U集合
  printf("k==%d\n",k);
  

  for(j=1;j<n;j++)
   if(flag[j]==0&&gm[k][j]<cost[j])//其中条件flag[j]==0是非常重要的
   {
    cost[j]=gm[k][j];//更新cost[]的最新值
    tree[j]=k;
    printf("    %d->%d\n",k,j);
   // printf("j==%d\n",j);
   }
 }
}

int main()
{  freopen("1.txt","r",stdin);
   printf("以序号为0的顶点出发生成最小生成树\n");
   int i,j,k,n,e,c;
   int tree[50],cost[50];
  
   scanf("%d%d",&n,&e);
   for(i=0;i<n;i++)
    for(j=0;j<n;j++)
     gm[i][j]=maxcost;

    for(k=1;k<=e;k++)
    {
     scanf("%d%d%d",&i,&j,&c);
     gm[i][j]=gm[j][i]=c;
    }

    prim(tree,cost,n);

    for(i=0;i<n;i++)
     printf("tree[%d]==%d\n",i,tree[i]);
   
    printf("\n");

    return 0;
}

 

//测试用例

/*7 10
0 1 50
0 2 60
1 3 65
1 4 40
2 3 52
2 6 45
3 4 50
3 5 30
3 6 42
4 5 70*/