最小生成树 Prim 算法HDU 2122 Ice_cream’s world III

Prim算法 ：

根据图中的顶点进行操作，而克鲁斯卡尔算法是先将边进行排序，从小到大加入集合。

1. 在图中任选一点作为开始节点，把这个顶点加入已经访问节点的集合。
2. 求出这样一点的相邻节点的距离。并找出最短的一条边。这是第一步，找到的第一条边。把这一个节点加入已经访问过的集合。
3. 在已经访问过的集合中，再次寻找最短的一条边。这一次还要考虑之前加入的顶点。因为之前的一个顶点到所有的距离已经求出。这一次就可以判断此点到其他点的距离，若距离小于之前对应的距离，则更新。

#include <iostream>
#include <cstdio>
#include <cstring>
#define INF 0x7fffffff
using namespace std;
int n, m;
int mymap[10010][10010];

void prim()
{
	int visited[10010], low[10010], ans = 0;
	memset(visited, 0, sizeof(visited));
	int pos = 0;
	for (int i = 1; i < n; i++)
	{
		low[i] = mymap[pos][i];
	}
	visited[pos] = 1;
	for (int i = 1; i < n; i++)<span style="white-space:pre">	</span>//这里一定是N - 1 次
	{
		int mincost = INF;
		for (int j = 0; j < n; j++)
		{
			if (!visited[j] && mincost > low[j])
			{
				mincost = low[j];
				pos = j;
			}
		}
		if (mincost == INF)
		{
			printf("impossible\n\n");
			return ;
		}
		visited[pos] = 1;
		ans += mincost;
		for (int i = 0; i < n; i++)
		{
			if (!visited[i] && mymap[pos][i] < low[i])
			{
				low[i] = mymap[pos][i];
			}
		}
	}
	printf("%d\n\n", ans);
}

int main()
{
	while (scanf("%d%d", &n, &m) != EOF)
	{
		for (int i = 0; i < n; i++)
		{
			for (int j = 0; j < n; j++)
			{
				if (i == j)
				{
					mymap[i][j] = 0;
				}
				else
				{
					mymap[i][j] = INF;
				}
			}
		}
		while (m--)
		{
			int a, b, c;
			scanf("%d%d%d", &a, &b, &c);
			if(mymap[a][b] > c)
            {
                mymap[a][b] = mymap[b][a] = c;
            }

		}
		prim();
	}
	return 0;
}