兔子与星空

学校poj平台上的题，考察最小生成树，这里用prim算法实现，思路见注释。

总时间限制: 1000ms 内存限制: 10000kB

描述
很久很久以前，森林里住着一群兔子。兔子们无聊的时候就喜欢研究星座。如图所示，天空中已经有了n颗星星，其中有些星星有边相连。兔子们希望删除掉一些边，然后使得保留下的边仍能是n颗星星连通。他们希望计算，保留的边的权值之和最小是多少？

输入
第一行只包含一个表示星星个数的数n，n不大于26，并且这n个星星是由大写字母表里的前n个字母表示。接下来的n-1行是由字母表的前n-1个字母开头。最后一个星星表示的字母不用输入。对于每一行，以每个星星表示的字母开头，然后后面跟着一个数字，表示有多少条边可以从这个星星到后面字母表中的星星。如果k是大于0，表示该行后面会表示k条边的k个数据。每条边的数据是由表示连接到另一端星星的字母和该边的权值组成。权值是正整数的并且小于100。该行的所有数据字段分隔单一空白。该星星网络将始终连接所有的星星。该星星网络将永远不会超过75条边。没有任何一个星星会有超过15条的边连接到其他星星（之前或之后的字母）。在下面的示例输入，数据是与上面的图相一致的。

输出
输出是一个整数，表示最小的权值和

样例输入
9
A 2 B 12 I 25
B 3 C 10 H 40 I 8
C 2 D 18 G 55
D 1 E 44
E 2 F 60 G 38
F 0
G 1 H 35
H 1 I 35

样例输出
216
 

#include<iostream>
#include<vector>
#define MAX 50
using namespace std;
bool visited[MAX];
int dist[MAX];//已访问点集到未访问点i的最短距离;
void prim(vector<vector<int>>& map,int &sum) {
	int n = map.size();
	for (int i = 0; i < n; i++) {//初始化两个数组
		dist[i] = 999;//不可达
		visited[i] = false;
	}
		
	visited[0] = true;//从第一个点开始访问
	int pos = 0;
	
	for (int i = 0; i < n - 1; i++) {
		for (int j = 0; j < n; j++) {
			if (map[pos][j] < dist[j]&&!visited[j])
				dist[j] = map[pos][j];
		}//更新dist[]
		int min = 999;
		for (int j = 0; j < n; j++) {
			if (dist[j] < min && !visited[j]) {
				min = dist[j];//找最小权值边
				pos = j;//以及对应终点
			}
		}
		sum += min;
		visited[pos] = true;
	}
}

int main() {
	int n;
	char id;//点的id
	int x; int y;
	int weight;//边权值
	int num;//相邻点的个数
	int sum = 0;
	cin >> n;
	//初始化图(用邻接矩阵储存)
	vector<vector<int>> map(n);
	for (int i = 0; i < n; i++) {
		map[i].resize(n);
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			map[i][j] = 999;//不可达
		}
	}

	for (int i = 0; i < n - 1; i++) {
		cin >> id >> num;
		x = id - 'A';
		for (int j = 0; j < num; j++) {
			cin >> id >> weight;
			y = id - 'A';
			map[x][y] = weight;
			map[y][x] = weight;
		}
	}
	prim(map,sum);
	cout << sum << endl;
}