Prim算法

#include<stdio.h>
#include<stdlib.h>

#define MAXVEX 100
#define INFINITY 65535

typedef struct{
	int vexs[MAXVEX];
	int arc[MAXVEX][MAXVEX];
	int numVertexes,numEdges;
}MGraph;

void creatMGraph(MGraph *G);
void MiniSpanTree_Prim(MGraph G);

void creatMGraph(MGraph *G)
{
	scanf("%d%d",&G->numVertexes,&G->numEdges);
	printf("%d %d\n",G->numVertexes,G->numEdges);
	
	for(int i = 0;i<G->numVertexes;i++)
	    scanf("%d",&G->vexs[i]);
	
	for(int i = 0;i<G->numVertexes;i++)
	printf("%d ",G->vexs[i]);
	printf("\n");
	
	for(int i = 0;i<G->numVertexes;i++)
	    for(int j = 0;j<G->numVertexes;j++)
	        G->arc[i][j] = INFINITY;
	        
	int i,j,w;
	for(int k = 0;k<G->numEdges;k++)
	{
		scanf("%d%d%d",&i,&j,&w);
		
		G->arc[i][j] = w;
		G->arc[j][i] = G->arc[i][j];
	}
	
	//printf("\n");
	for(int i = 0;i<G->numVertexes;i++)
	{
		for(int j = 0;j<G->numVertexes;j++)
	    {
	 	    if(G->arc[i][j]!=INFINITY)
		    printf("%2d ",G->arc[i][j]);
		    else
		    printf("&  ");
	    }
	    printf("\n");
	}
	printf("\n");
}

void MiniSpanTree_Prim(MGraph G)
{
	int min, i, j, k;
	int adjvex[MAXVEX];          //保存相关顶点下标                                                       第 4 行 
	int lowcost[MAXVEX];         //保存相关顶点间边的权值

	lowcost[0] = 0;               //初始化第一个权值为 0，即 v0 加入生成树
	                              //lowcost 的值为 0 在这里就是此下标的顶点已经加入生成树
	adjvex[0] = 0;                //初始化第一个顶点下标为 0
	for (i = 1; i < G.numVertexes; i++)         //循环除下标为 0 外的全部顶点
	{
		lowcost[i] = G.arc[0][i];       //将 v0 顶点与之有边的权值存入数组
		adjvex[i] = 0;                  //初始化都为 v0 的下标
	}

	for (i = 1; i < G.numVertexes; i++)
	{
		min = INFINITY;              //初始化最小权值为无穷
		                             //通常设置不可能的大数字如32767，65535等作为无穷
		j = 1;                       
		k = 0;

		while (j < G.numVertexes)             //循环全部顶点
		{
			if (lowcost[j] != 0 && lowcost[j] < min)
			{
				//如果权值不为0且权值小于min
				min = lowcost[j];           //则让当前权值成为最小值
				k = j;                      //将当前最小值的下标存入 k
			}
			j++;
		}
		
		printf("(%d,%d)", adjvex[k], k);       //打印当前顶点边中权值最小边
		
		lowcost[k] = 0;           //将当前顶点的权值设置为 0，表示此顶点已经完成任务

		for (j = 1; j < G.numVertexes; j++)        //循环所有顶点 -- 这里是为了更新 lowcost 数组的值，在经历上一次循环后，已经多纳入了一个顶点， 
		{                                          //这时要找的下一个顶点可能与当前纳入的顶点，和与这个顶点相连的那个顶点有关，此时对所有顶点做遍历
		                                           //如果与 刚纳入的顶点 k 之间的权值 G.arc[k][j] 小于之前相连的那个顶点的对应权值 lowcost[j]，则用新的
												   //小的权值来替换，之前长的权值作废，并且把这个顶点纳入到基准顶点数组 adjvex 中 
			if (lowcost[j] != 0 && G.arc[k][j] < lowcost[j])
			{
				//若下标为k顶点各边权值小于此前这些顶点未被加入生成树权值
				lowcost[j] = G.arc[k][j];        //将较小权值存入 lowcost
				adjvex[j] = k;                   //将下标为 k 的顶点存入 adjvex
			}
		}
	}
}

int main(void)
{
    freopen("最小生成树测试数据.txt","r",stdin);
	MGraph *G = (MGraph *)malloc(sizeof(MGraph));
	creatMGraph(G);
	MiniSpanTree_Prim(*G);
	
	
	for(int i = 0;i<MAXVEX;i++)
	free(G->arc[i]);
	
	free(G->vexs);
	free(G);
	
	return 0;
}