Prim算法的实现

好吧，这是我写的Prim算法，直接上代码：

/*测试数据

6 10
1 2 6
1 3 1
1 4 5
2 3 5
3 4 5
2 5 3 
3 5 6
3 6 4
4 6 2
5 6 6

*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define N 100  //定义最大顶点数N

const int MaxInt = 99999999;

int map[N][N], low[N], visited[N], n;  //map数组存储图的邻接矩阵形式的权值
                            //low数组记录每2个结点间最小权值，visited数组标记是否已被访问

int prim()
{
	int i, j, pos, min;   // pos为位置
	int result = 0;

	memset(visited, 0, sizeof(visited));
	visited[1] = 1, pos = 1;  // 从某点开始， 分别标记和记录该点


	for (i = 1; i <= n; i++)// 第一次给low数组赋值
		if(i != pos)
			low[i] = map[pos][i];
		
	printf("路径为：%d->", pos);

	for (i = 1; i < n; i++)// 再运行n-1次(给low数组赋值的操作)
	{
		// 找出最小权值并记录位置
		min = MaxInt;  
		for (j = 1; j <= n; j++)
			if(visited[j] == 0 && min > low[j])
			{
				min = low[j];
				pos = j;
			}
		printf("%d->", pos);
		result += min;// 最小权值累加
		visited[pos] = 1; // 标记该点

		//更新权值
		for (j = 1; j <= n; j++)
			if(visited[j] == 0 && low[j] > map[pos][j])
				low[j] = map[pos][j];
	}

	return result;
}

int main()
{
	int i, j, m, s, e, w;

	while(scanf("%d %d", &n, &m) != EOF)
	{
		// 所有权值初始化为最大
		for(i = 1; i <= n; i++)
			for(j = 1; j <= n; j++)
				map[i][j] = MaxInt;

		for (i = 0; i < m; i++)
		{
			scanf("%d %d %d", &s, &e, &w);
			map[s][e] = map[e][s] = w;
		}
			
		printf("\n总权值 = %d\n", prim());
	}

	return 0;
}