Prim法求最小生成树

     最小生成树：对于一个带权的无向连通图，其每个生成树所有边上的权值之和可能不同，我们把所有边上权值之和最小的生成树称为图的最小生成树。

普利姆法算法：对于一个图来说，要设立二个数组，一个数组用来记录权值，数组值来自开始顶点的权值，另一个数组

记录每次加入的顶点，找到最小权值的顶点，加入到这个数组。因为有新顶点的加入，所以可以更新权值数组的值，

不断加入顶点，不断更新。就能找到最小生成树。

图例：

代码：

#include<stdio.h>
#include<stdlib.h>
#include <malloc.h>
#include <string.h>
#define MAX 20
#define INF 100      // 此处修改最大值
#define nLENGTH(a)  (sizeof(a)/sizeof(a[0]))
#define eLENGTH(a) ( sizeof(a) / sizeof(char) )/ ( sizeof(a[0]) / sizeof(char) )
typedef struct _graph
{
    char vexs[MAX];       // 顶点集合
    int vexnum;           // 顶点数
    int edgnum;           // 边数
    int matrix[MAX][MAX]; // 邻接矩阵
}Graph, *PGraph;
// 边的结构体 
typedef struct _EdgeData
{
    char start; // 边的起点
    char end;   // 边的终点
    int weight; // 边的权重
}EData;
//指向节点的位置
int point_node(PGraph g,char c)
{
	for(int i=0;i<g->vexnum;i++)
	{
		if(g->vexs[i]==c)
		{
			return i;
		}
	}
	return -1;
}
PGraph create_graph(int b[][3],char a[],int n,int e)
{
	char c1,c2; //边的2个顶点
	PGraph g; //矩阵
	g=(PGraph)malloc(sizeof(Graph));
	//memset()第一个参数 是地址，第二个参数是开辟空间的初始值，第三个参数是开辟空间的大小
	memset(g, 100, sizeof(Graph));
	printf("顶点个数:\n");//顶点数
	g->vexnum=n;
	printf("%d\n",g->vexnum);
	printf("边个数:\n");//边数
	g->edgnum=e;
	printf("%d\n",g->edgnum);
	//初始化顶点
	for(int j=0;j<g->vexnum;j++)
	{
		g->vexs[j]=a[j];
	}
	for(int i=0;i<g->edgnum;i++)
	{
		int p1,p2;
		c1=char(b[i][0]);
		c2=char(b[i][1]);
		p1=point_node(g, c1);
		p2=point_node(g, c2);
		if (p1==-1 || p2==-1)
        {
            printf("input error: invalid edge!\n");
            free(g);
            continue;
        }
		g->matrix[p1][p2]=b[i][2];
		g->matrix[p2][p1]=b[i][2];
		
	}
	return g;
}
//普利姆法求最小生成树 
void PrimTree(PGraph g,char a) //从start顶点开始求解
{
	int start=point_node(g,a);//得到顶点对应的号
	int index=0;
	char prim[MAX];    //存放每次刷新后得到的顶点
	int weight[MAX]={0}; //存放每次刷新顶点的权值
	for(int i=0; i<g->vexnum;i++)
	{
		weight[i]=g->matrix[start][i]; //取得开始顶点的所有权值
	}
	weight[start]=0;
	prim[index++]=a;
	for(i=0;i<g->vexnum;i++)
	{
		int k; //用来存放最小权值，可以找到它所对应的顶点
		int min=INF;
		//找到最小权值的顶点，并添加入prim数组
		for(int j=0; j<g->vexnum;j++)
		{
			if(weight[j]!=0 && weight[j]<min)
			{
				min=weight[j];
				k=j;
			}
		}
		printf("\n");
		//找到k后，要把start,k 之间的权值设为0；
		weight[k]=0;
		prim[index++]=g->vexs[k];
		//更新weight权值，把k顶点的权值加入到
		for( j=0; j<g->vexnum;j++)
		{
			if(weight[j]!=0 && g->matrix[k][j]<weight[j])
			{
				weight[j]=g->matrix[k][j]; //现在的 weight组 的值 即为顶点 start，k 的权值总和，下次
										   //寻找最小权值在这个组里找
			}
		}

	}
	//打印最小生成树的顺序	 ;
	for(int m=0;m<g->vexnum;m++)
	{
		printf("%c\t",prim[m]);
	}
}



//测试
int main()
{
	int i,j;
	PGraph gp;
	//测试用例


	char a[]={'A', 'B', 'C', 'D', 'E', 'F', 'G'};
	int b[][3]={
        {'A', 'B',12}, 
        {'A', 'F',16}, 
        {'A', 'G',14}, 
        {'B', 'F',7}, 
        {'B', 'C',10}, 
        {'C', 'F',6}, 
        {'C', 'E',5},
		{'C', 'D',3},
		{'D', 'E',4},
		{'E', 'F',2},
		{'E', 'G',8},
		{'F', 'G',9}}; 

	//测试用例

	int n=nLENGTH(a);
	int e=eLENGTH(b);
	gp=create_graph(b,a,n,e);

	//打印矩阵
	for (i = 0; i < gp->vexnum; i++)
    {
        for (j = 0; j < gp->vexnum; j++)
            printf("%d ", gp->matrix[i][j]);
        printf("\n");
    }
	//普利姆法求最小生成树
	PrimTree(gp,'A');

	return 0;

}