【C语言数据结构】最小生成树Prim算法 | 原理和代码

一、代码

#include <stdio.h>
#include <stdlib.h>
#define maxvex 20
#define INFINITY 65535
typedef struct
{
    char vex;
    int arc[maxvex][maxvex];
    int numvex,numedge;
}MGraph;

typedef struct shortedge     //最短边的顶点集合
{
    int lowcost;   //边的权重
    int adjvex;     //记录顶点
}shortEdge[maxvex];

void creatMGraph(MGraph *G)
{
    int i,j,k,w;
    printf("输入顶点数和边数\n");
    scanf("%d %d",&G->numvex,&G->numedge);

    for(i=0;i<G->numvex;i++)           //初始化临阶矩阵为无穷
        for(j=0;j<G->numvex;j++)
            G->arc[i][j]=INFINITY;
                                        //vex赋值省略
    for( k=0;k<G->numedge;k++)
    {
        printf("输入边(vi,vj)的下标i j以及权重w，用逗号隔开\n");
        scanf("%d %d %d",&i,&j,&w);
        G->arc[i][j]=w;
        G->arc[j][i]=w;
    }

}

void Prim(MGraph G,int start)
{
    shortEdge s;
    int i,minn,k;
    for(i=0;i<G.numvex;i++)
    {
        s[i].lowcost=G.arc[start][i];
        s[i].adjvex = start;
    }
    s[start].lowcost= 0;           //把起点放入集合中

    //上面是初始化起点,下面是处理其他点

    for(i=0;i<G.numvex-1;i++)     //（1）寻找k值为最短边的新定点，minn为边长
    {
        int j=0;
        minn=INFINITY;
        while(j<G.numvex)
        {
            if(s[j].lowcost!=0 && s[j].lowcost<minn)
            {
                minn = s[j].lowcost;
                k=j;
            }
            j++;
        }


        printf("(%d,%d) %d\n",s[k].adjvex,k,minn);    //(2)输出
        s[k].lowcost=0;                            //（3）加入集合

        for(int j=0;j<G.numvex;j++)             //（4）加入新的点后重新确定 新点集合的最短边矩阵
        {
            if(s[j].lowcost!=0 && G.arc[k][j]<s[j].lowcost)
            {
                s[j].lowcost=G.arc[k][j];
                s[j].adjvex=k;
            }
        }

    }
}
int main()
{
    MGraph G;
    creatMGraph(&G);
    Prim(G,0);
    return 0;
}

结果：

二、原理

1、概念

（1）最小生成树：包含无向连通G全部顶点的极小连通子图
边数 n-1
（2）Prim 算法：从一个点出发，依次加入点形成点集U （全集为V）
Kruskal 算法：从边出发，将点集合并，避免形成环

2、实现方法

（1）通过 shortEdge[n]结构体
包含了 adjvex（顶点） lowcost（边的权值）
不用也行，可以分别定义两个数组
（2）首先对lowcost赋起点的边值，找到最短的边，记下顶点的下标；（初始化）
把这个点加入U集合中 --> lowcost=0；
遍历这个顶点的边，修改lowcost数组
重复这个动作，把V集合的点都加入U集合为止。
如图，这是起点为0 时，加入第一个点5 的过程。

3、注意

（1）起点不同，最小生成树是一样的
（2）时间复杂度 O（n*2），与顶点个数相关。