Prim最小生成树

思路

先从一个顶点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;
}