最小生成树:对于一个带权的无向连通图,其每个生成树所有边上的权值之和可能不同,我们把所有边上权值之和最小的生成树称为图的最小生成树。
普利姆法算法:对于一个图来说,要设立二个数组,一个数组用来记录权值,数组值来自开始顶点的权值,另一个数组
记录每次加入的顶点,找到最小权值的顶点,加入到这个数组。因为有新顶点的加入,所以可以更新权值数组的值,
不断加入顶点,不断更新。就能找到最小生成树。
图例:
代码:
#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;
}