//主要源代码如下
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <malloc.h>
#define INF 9999999
#define MAX_NAME 3
#define VERTEX_MAX_NUM 100
typedef char VertexType[MAX_NAME];
typedef int VRType;
typedef struct{
int weight;
}adjMatrix[VERTEX_MAX_NUM][VERTEX_MAX_NUM];//邻接矩阵
struct MGraph{
VertexType vex[VERTEX_MAX_NUM];//顶点向量
adjMatrix arcs;//邻接矩阵
int vrtnum,arcnum;//图的当前顶点数和边数
};
typedef struct mini {
VertexType adjvex; //保存顶点名字
VRType lowcost;
}minside[VERTEX_MAX_NUM];
int LocateVex(MGraph G,VertexType u){
//传入一个顶点的名字,定位它在图中的位置。否则返回-1。(返回数组下标)
int i;
for(i=0;i<G.vrtnum;i++)
if(strcmp(u,G.vex[i])==0)
return i;
return -1;
}
void creatUDG(MGraph &G){ //构建图。
int i,j,k,w;
VertexType va,vb;
printf("请输入无向图G的顶点数、边数:");
scanf("%d %d",&G.vrtnum,&G.arcnum);
printf("请输入%d个顶点的值:\n",G.vrtnum);
for(i=0;i<G.vrtnum;i++) //构造顶点向量,(其实就是将顶点的名字换成数字)
scanf("%s",G.vex[i]);
for(i=0;i<G.vrtnum;i++) //初始化邻接矩阵,都赋为无穷
for(j=0;j<G.vrtnum;j++)
G.arcs[i][j].weight=INF;
printf("请输入%d条边的顶点1 顶点2 权值(以空格为间隔):\n",G.vrtnum);
for(k=0;k<G.arcnum;k++){
scanf("%s%s%d%*c",va,vb,&w);
i=LocateVex(G,va);//找顶点对应的数字名字
j=LocateVex(G,vb);
G.arcs[i][j].weight=G.arcs[j][i].weight=w;
//给存在关系的点在邻接矩阵中赋值
}
}
void DisplayArc(MGraph g){
printf("建立的邻接矩阵如下:\n ");
for(int i=0;i<g.vrtnum;i++)
printf("%7s",&g.vex[i]);
printf("\n");
for(int i=0;i<g.vrtnum;i++){
printf("%s",g.vex[i]);
for(int j=0;j<g.vrtnum;j++){
if(g.arcs[i][j].weight==INF)
printf(" ∞");
else
printf("%7d",g.arcs[i][j].weight);
}
printf("\n");
}
}
int mininum(minside SZ,MGraph G){
int i=0,j,k,min;
while(!SZ[i].lowcost)
i++;
//得到第一个不为0的值,然后用这值来跟后面的lowcost进行比较
min=SZ[i].lowcost;
k=i;
for(j=i+1;j<G.vrtnum;j++){//得到大于0的最小值
if(SZ[j].lowcost>0&&SZ[j].lowcost<min){
min=SZ[j].lowcost;
k=j;
}
}
return k;
}
void MiniSpanTree_PRIM(MGraph G,VertexType u){
int i,j,k,allCost=0;
minside closedge;//记录从U到V-U具有最小代价的边。
k=LocateVex(G,u);
for(j=0;j<G.vrtnum;++j){//初始化辅助数组
strcpy(closedge[j].adjvex,u);//赋值u的名字到辅助数组中
closedge[j].lowcost=G.arcs[k][j].weight;//所有点到初始值的权值
}
closedge[k].lowcost=0;//初始点到自己的权值为0,U={u}。
printf("最小代价生成树的各条边为:\n");
int t=k;
for(i=1;i<G.vrtnum;++i){
k=mininum(closedge,G);
printf("(%s-%s) %d\n",closedge[k].adjvex,G.vex[k],G.arcs[t][k].weight);
allCost+=G.arcs[t][k].weight;
t=k;
closedge[k].lowcost=0; //第k个顶点并入U集
for(j=0;j<G.vrtnum;++j){ //新顶点并入U集后重新选择最小边
if(G.arcs[k][j].weight<closedge[j].lowcost){
strcpy(closedge[j].adjvex,G.vex[k]);
closedge[j].lowcost=G.arcs[k][j].weight;
}
}
}
printf("最小生成树的总权值为:%d\n",allCost);
}
int main()
{
MGraph g;
creatUDG(g);
DisplayArc(g);
MiniSpanTree_PRIM(g,g.vex[0]);
return 0;
}
/*
测试样例:
v0 v1 v2 v3 v4 v5 v6
v0 v1 28
v1 v2 16
v2 v3 12
v3 v4 22
v4 v5 25
v5 v0 10
v1 v6 14
v3 v6 18
v4 v6 24
*/