练习四 1006

概述：也是修路问题，给定一个地图，求修路的最小花费。

思路：最小生成树的题目，这次使用的Prim算法，选一个节点放到数组中，找出和他相邻的最便宜的节点，再把这个节点放到数组中，然后继续重复这个步骤，金钱也不断的累加，最后输出金钱数目即可。

感想：感觉前6个题，用一套代码改改就行。。。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
const long long INF = 0x7fffffff;
using namespace std;

int map[30][30], visit[30], distance[30];
int n;

void prim()
{
	for(int i=1; i<=n; i++)
	{
		distance[i] = map[1][i];
	}
	distance[1] = 0; visit[1] = 1;
	int sum = 0, pos;
	for(int i=1; i<n; i++)
	{
		int tmp = INF;
		for(int j=1; j<=n; j++)
		{
			if(!visit[j] && distance[j]!=-1 && distance[j]<tmp)
			{
				tmp = distance[j];
				pos = j;
			}
		}
		sum += tmp;
		visit[pos] = 1;
		for(int i=1; i<=n; i++)
			if(!visit[i] && (map[pos][i] < distance[i] || distance[i] == -1) && map[pos][i] != -1)
				distance[i] = map[pos][i];
	}
	cout<<sum<<endl;
}

int main()
{
	char a[3], b[3];
	int m, w;
	while(cin>>n&&n)
	{
		memset(map, -1, sizeof(map));
		memset(distance, -1, sizeof(distance));
		memset(visit, 0, sizeof(visit));
		for(int i=0; i<n-1; i++)
		{
			scanf("%s %d", a, &m);
			for(int i=0; i<m; i++)
			{
				scanf("%s %d", b, &w);
				map[a[0]-64][b[0]-64] = w;
				map[b[0]-64][a[0]-64] = w;
			}
		}
		prim();
	}
	return 0;
}