prim算法（最短生成树）

prim算法是基于贪心的思想寻求最短生成树，使用三个数组来实现的，第一个二维数组存点到邻点的边长，一个一维数组ju[i]存树任何一点到i点的最短路径，bool数组防止重复累加，

思路是从起点出发，遍历起点能达到的点的距离，将其中最短的距离到达的点加入集合，更新集合能达到的点的最短距离，重复上述操作，直达所有的点都在集合中，结束。

 

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 505, INF = 0x3f3f3f3f;
int n, G[N][N];
//vector<int>x(N+1,INF);
bool vis[N];//判断点是否使用
int tu[N][N];//将图存入邻接矩阵
int ju[N];//点到集合的最短距离
int m;
int prim()
{
	memset(ju, 0x3f, sizeof(ju));
	int res = 0;//所有边的权值和
	for (int i = 0; i < m; ++i)//当i=0时是第一次向ju中读入数据
	{
		int t = -1;
		for (int j = 1; j <= m; ++j)
		{
			if (!vis[j] && (t == -1 || ju[t] > ju[j]))
			{
				t = j;
			}
		}
		if (i && ju[t] == INF)//说明所有边不是联通的
		{
			return INF;
		}
		if (i)
		{
			res += ju[t];
		}
		vis[t] = true;
		for (int j = 1; j <= m; ++j)
		{
			ju[j] = min(ju[j], tu[t][j]);
		}
	}
	return res;
}
int main()
{
	int n, a, b, c;
	cin >> m >> n;
	memset(tu, INF, sizeof(tu));
	for (int i = 0; i < n; ++i)
	{
		cin >> a >> b >> c;
		tu[a][b] = tu[b][a] = min(tu[a][b], c);//无向图, 建立双向边
	}
	int t = prim();
	cout << t << endl;
	return 0;
}