【描述】:
采用邻接矩阵方式储存图,创建图,生成最小生成树。
【输入】:
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;
}