数据结构 — 图 之 MST（最小生成树 — prim算法 ）

【描述】：

采用邻接矩阵方式储存图，创建图，生成最小生成树。

【输入】：

7 9
0 1 2 3 4 5 6
0 1 28
0 5 10
1 2 16
1 6 14
2 3 12
3 4 22
3 6 18
4 5 25
4 6 24

【输出】：

0,5权值: 10 | 5,4权值: 25 | 4,3权值: 22 | 3,2权值: 12 | 2,1权值: 16 | 1,6权值: 14 | 

/*
7 9
0 1 2 3 4 5 6
0 1 28
0 5 10
1 2 16
1 6 14
2 3 12
3 4 22
3 6 18
4 5 25
4 6 24
*/
#include<iostream>

using namespace std;

/*宏定义*/
#define INFINITY 65535
#define MAX_NUM 100
#define EleType int
#define EdgeType int

/*图的结构体*/
typedef struct {
    EleType vertex[MAX_NUM];    //顶点表
    EdgeType arc[MAX_NUM][MAX_NUM];  //邻接矩阵
    int VertexNum;  //顶点数
    int ArcNum; //边数
}GraphType,*GraphPointer;

//声明图
GraphType graph;

/*邻接矩阵的创建*/
void CreateGraph(){
    int w;  //权值
    int n,k;
    //输入顶点数和边数
    cin>>graph.VertexNum>>graph.ArcNum;
    //输入所有顶点
    for(int i = 0; i<graph.VertexNum; i++){
        cin>>graph.vertex[i];
    }
    //初始化邻接矩阵
    for(int j = 0; j<graph.VertexNum; j++){
        for(int k = 0; k<graph.VertexNum; k++){
            if(j == k){
                graph.arc[j][k] = 0;
            }else{
                graph.arc[j][k] = INFINITY;
            }
        }
    }
    //输入邻接矩阵中的边
    for(int i = 0; i<graph.ArcNum; i++){
        //输入顶点j、k、边的权值w
        cin>>n>>k>>w;
        graph.arc[n][k] = w;
        //无向图邻接矩阵对称原则
        graph.arc[k][n] = w;
    }
    
}

/*prim*/
void prim(){
    int n,min;  //n是当前最小值的下标（根据lowcost数组得出 ）
    int adjver[MAX_NUM];    //所有vertex 与 已经加入到MST中，可以和其组成min距离的顶点相连
    int lowcost[MAX_NUM];   //所有vertex 与 MST的最小距离（权值）
    adjver[0] = 0;  //相连
    lowcost[0] = 0; //更新权值

    //将第一顶点v0加入到MST中的第一轮更新
    for(int i = 1; i<graph.VertexNum; i++){
        adjver[i] = 0;  //MST中只有 v0
        lowcost[i] = graph.arc[0][i];   //将第一行的所有权值放入该数组，因为MST现在只有v0
    }

    //构建MST
    for(int i = 1; i<graph.VertexNum; i++){
        min = INFINITY;
        n = 0;

        //根据lowcost数组得出 n
        for(int j = 1; j<graph.VertexNum; j++){
            if(lowcost[j] != 0 && lowcost[j]<min){
                min = lowcost[j];
                n = j;
            }
        }

        //输出
        cout<<adjver[n]<<","<<n<<"权值: "<<graph.arc[adjver[n]][n]<<" | ";
        //lowcost = 0 说明已经加入 MST
        lowcost[n] = 0;

        //根据 从n 开始的所有边的权值大小去更新 lowcost数组
        for(int k = 0; k<graph.VertexNum; k++){
            if(lowcost[k] != 0 && graph.arc[n][k] < lowcost[k]){
                lowcost[k] = graph.arc[n][k];
                adjver[k] = n;  //k 与 n（已经是MST中的元素）相连
            }
        }

    }
    cout<<endl;
}

int main(){

    CreateGraph();
    cout<<endl;
    prim();

    return 0;
}