【数据结构】Prim算法求最小生成树（贪心）

时间复杂度：O(n²)

#include<iostream>
#define n 6
#define INF 999
using namespace std;

// 定义辅助数组的元素类型 
typedef struct{
	int lowcost;  // 最短的值 
	int adjvex;  // 邻接点 
}Element;

void Prim(int arc[n][n],int w) 
{
	Element shortEdge[10]; // 存储后端最短边集，假设最多10个顶点 
	for(int i=0;i<n;i++){
		shortEdge[i].lowcost = arc[w][i];
		shortEdge[i].adjvex = w;
	}
	shortEdge[w].lowcost = 0; // 将顶点 w加入集合U
	int k,min;
	for(int i=0;i<n-1;i++){
		min = 500; // 假设权值均小于100
		for(int j=0;j<n;j++) { // 寻找最短边的邻接点k 
			if((shortEdge[j].lowcost!=0) && (shortEdge[j].lowcost<min)){
				min = shortEdge[j].lowcost;
				k = j; // 最短边对应的点 
			} 
		}
		cout<<shortEdge[k].adjvex<<"--"<<k<<"  "<<min<<endl; // 输出最小生成树的边
		shortEdge[k].lowcost = 0;  // 将顶点k加入集合U中 
		// 调整shortEdge[n] 
		for(int j=0;j<n;j++){  
			if(arc[k][j]<shortEdge[j].lowcost){
				shortEdge[j].lowcost = arc[k][j];
				shortEdge[j].adjvex = k;
			}
		} 
	} 
}

int main()
{
	int arc[n][n] = {{INF, 34, 46,INF,INF, 19},
	                 { 34,INF,INF,INF, 12,INF},
					 { 46,INF,INF, 17,INF, 25},
					 {INF,INF, 17,INF, 38, 25},
					 {INF, 12,INF, 38,INF, 26},
				     { 19,INF, 25, 25, 26,INF}};
	Prim(arc,0);
	return 0;
}