prim（普里姆）算法求图的最小生成树

prim算法求最小生成树：
/*
题目 1705: 算法7-9：最小生成树

输入
输入的第一行包含一个正整数n，表示图中共有n个顶点。其中n不超过50。
以后的n行中每行有n个用空格隔开的整数，对于第i行的第j个整数，如果不为0，则表示第i个顶点和第j个顶点有直接连接且代价为相应的值，0表示没有直接连接。当i和j相等的时候，保证对应的整数为0。
输入保证邻接矩阵为对称矩阵，即输入的图一定是无向图，且保证图中只有一个连通分量。
输出
只有一个整数，即最小生成树的总代价。请注意行尾输出换行。
样例输入
4
0 2 4 0
2 0 3 5
4 3 0 1
0 5 1 0
样例输出
6
*/

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXVAL 32767
#define MAXN 100
int visited[MAXN];
int edge[MAXN][MAXN];
int min_dis[MAXN]; //记录现有顶点到其它顶点的最小距离

int sum = 0;
int n;
int count_num = 0;
void prime(int k)
{
	visited[k] = 1; //访问当前顶点
	for (int j = 1; j <= n; j++) 
	{
		min_dis[j] = edge[k][j]; //初始化最小距离数组（初始时刻顶点k加入顶点集合，min_dis记录了从该顶点到其余所有顶点的最小距离）
		if (edge[k][j] == 0) //不可达
		{
			min_dis[j] = MAXVAL;
		}
	}
	int index = 1; //记录最小值的结点
	for (int i = 1; i < n; i++) //遍历所有的顶点,选择n-1条边 
	{
		int min_temp = MAXVAL;
		for (int j = 1; j <= n; j++)
		{
			if (min_dis[j] < min_temp && visited[j] == 0)
			{
				min_temp = min_dis[j];
				index = j;
			}
		}
		visited[index] = 1;
		sum += min_dis[index];
		for (int k = 1; k <= n; k++) //更新当前顶点集合(已经连接的顶点)中到其它未加入顶点集合（未连接的顶点）的最小路径
		{
			if (visited[k] == 1)
				continue;
			min_dis[k] = min(min_dis[k], edge[index][k]); 
		}
	}

}

int main()
{
	scanf("%d", &n);
	int val = 0;
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			scanf("%d", &edge[i][j]); 
			if (edge[i][j] == 0)
				edge[i][j] = MAXVAL;
		}
	}
	prime(1); //从顶点1开始遍历
	printf("%d\n", sum);
	return 0;
}