prim 算法 最小生成树

#include <iostream>
using namespace std;


int const M = 65536;


void prim(int graph[][4],int n){
int* adjnode = new int[n]; 
int* lowcost = new int[n];
//所有已选节点中到每一个节点最短权的节点数组,例如[0,0,2,2,0,2]，表示到下一个节点最近的点
adjnode[0] = 0;
//
lowcost[0] = 0;
//初始化，读取第一行数据，即第一个节点到各节点权
for(int i=1;i!=n;i++){
lowcost[i] = graph[0][i];
//初始化所有数据都是第一个节点
adjnode[i] = 0;
}


for(int i=1;i!=n;i++){
int minIndex = 0;
int min = M;
for(int j=1;j!=n;j++){
if(lowcost[j]!=0&&min>lowcost[j]){
//在已知点集合的所有边中最小权
min = lowcost[j];
//和已知节点构成最小权的节点
minIndex = j;
}
}
cout<<"节点："<<adjnode[minIndex]<<","<<minIndex<<endl;
//给lowcost赋值0表示这个点已经被选取过,避开已选节点
//可以理解成按列排除
lowcost[minIndex] = 0;

for(int j=0;j!=n;j++){
//选择已选节点到未选节点最小距离的点
if(lowcost[j]!=0&&graph[minIndex][j]<lowcost[j]){
lowcost[j] = graph[minIndex][j];
adjnode[j] = minIndex;
}
}
}
}


int main(){
int graph[4][4] = {{1,2,3,4},{M,M,M,M},{M,M,M,M},{M,M,M,M}};
prim(graph,4);
return 0;
}