思路
先从一个顶点u出发,找u的邻接顶点v,如果(u,v)权值最小且v顶点不在生成树顶点集合中(防止出现回路),则把边(u,v)存到最小生成树中;否则该顶点已经在生成树顶点集合中,舍弃该边,找权值次小的另一条边。然后,从顶点v出发,找下一条不会出现回路的边(v,w),直到找出(n-1)条边
例题
输入描述
首先输入图中顶点个数和边的条数;
再输入顶点的信息(字符型);
再输入各边及其权值。
输出描述
输出从编号为0的顶点开始的Prim算法最小生成树中的各边及其权值,每条边及其权值占一行。
输入样例
5 7
A B C D E
0 1 6
0 2 2
0 3 1
1 2 4
1 3 3
2 4 6
3 4 7
输出样例
A D 1
A C 2
D B 3
C E 6
#include <iostream>
using namespace std;
const int MaxSize = 10;
const int INF = 32767;
class MGraph
{
public:
MGraph(char a[], int n, int e);
void Prim();
private:
char vertex[MaxSize];
int arc[MaxSize][MaxSize];
int vertexNum, arcNum;
};
MGraph::MGraph(char a[], int n, int e)
{
//write your code.
int i,j,weight;
vertexNum=n,arcNum=e;
for(int i=0;i<vertexNum;i++){
vertex[i]=a[i];
}
for(int i=0;i<arcNum;i++){
for(int j=0;j<arcNum;j++){
arc[i][j]=INF;
}
}
for(int k=0;k<arcNum;k++){
cin>>i>>j>>weight;
arc[i][j]=weight;
arc[j][i]=weight;
}
}
int MinEdge(int lowcost[], int vertexNum)
{
//write your code.
int min=INF;
int k=0;
for(int i=0;i<vertexNum;i++){
if(min>lowcost[i]&&lowcost[i]!=0){
min=lowcost[i];
k=i;
}
}
return k;
}
void MGraph::Prim()
{
int k;
//write your code.
int lowcost[MaxSize];
int adjvex[MaxSize];
for(int i=0;i<vertexNum;i++){
lowcost[i]=arc[0][i];
adjvex[i]=0;
}
lowcost[0]=0;
for(int i=1;i<vertexNum;i++){
k=MinEdge(lowcost,vertexNum);
cout<<vertex[adjvex[k]]<<" "<<vertex[k]<<" "<<lowcost[k]<<endl;
lowcost[k]=0;
for(int j=0;j<vertexNum;j++){
if(arc[k][j]<lowcost[j]){
lowcost[j]=arc[k][j];
adjvex[j]=k;
}
}
}
}
int main()
{
int n = 0;
int e = 0;
cin >> n >> e;
char p[MaxSize];
int i = 0;
for (i=0; i<n; i++)
{
cin >> p[i];
}
MGraph MG(p, n, e);
MG.Prim();
return 0;
}