prim算法求最小生成树

描述

使用prim算法求某图的最小生成树的边的权值输出的序列。例如下图的最小生成树的权值输出序列为1 4 2 5 3，要求从V1顶点开始生成最小生成树。

输入
若干行整数
第一行为两个整数，分别为图的顶点数和边数
第二行开始是该图的邻接矩阵，主对角线统一用0表示，无直接路径的两点用100来表示（保证各边权值小于100）
输出
若干用空格隔开的整数
样例输入

6 10
0 6 1 5 100 100
6 0 5 100 3 100
1 5 0 5 6 4
5 100 5 0 100 2
100 3 6 100 0 6
100 100 4 2 6 0

样例输出

1 4 2 5 3

代码

#include<bits/stdc++.h>
using namespace std;
int path[1000],idx,n,m,mapp[200][200],dis[1000];
bool vis[1000];
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&mapp[i][j]);
		}
	}
	memset(dis,127,sizeof(dis));
	dis[1]=0;
	for(int i=1;i<=n;i++){
		int u=0;
		for(int j=1;j<=n;j++){
			if(!vis[j]&&dis[u]>dis[j])u=j;
		}
		vis[u]=1;
		path[++idx]=dis[u];
		for(int v=1;v<=n;v++){
			if(!vis[v])dis[v]=min(mapp[u][v],dis[v]);
		}
	}
	for(int i=2;i<=idx;i++){
		cout<<path[i]<<" ";
	}
	return 0;
}