prim算法

裸题传送门

#include <iostream>
#include <stdio.h>
#define MAXN 110
#define INTFY 1<<20
using namespace std;
bool book[MAXN];//标记数组，判断改点是否走过
int dist[MAXN];//记录已经走过的点里到该点的最短距离
int mmp[MAXN][MAXN];//邻接矩阵
int main(void)
{
	int n;
	int min_sum = 0;

	scanf("%d", &n);
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
		{
			int temp;

			scanf("%d", &temp);
			mmp[i][j] = (temp != -1) ? temp : INTFY;
		}//输入矩阵
	book[0] = true;//从第一个点开始吧（从任意点开始都是允许的）
	for (int i = 0; i < n; i++)
		dist[i] = mmp[0][i];//第一个点到周围点的最短距离们
	while (true)
	{
		int position;
		int min_num = INTFY;

		for (int i = 0; i < n; i++)
			if (!book[i] && min_num > dist[i])
			{//寻找最短距离
				min_num = dist[i];
				position = i;
			}
		if (min_num == INTFY)break;//如果找不到了，那一定是找完了
		min_sum += min_num;//加上这个最短距离
		book[position] = true;//走到这个位置了
		for (int i = 0; i < n; i++)//对走到的当前点的周围的点进行一次搜索
			if (!book[i] && dist[i] > mmp[position][i])
				dist[i] = mmp[position][i];//如果出现一个没有走过并且更短的路就更新数据
	}
	printf("%d\n", min_sum);
	//system("pause");

	return 0;
}