HDU 1301 %*C显示威力

最小生成树

#include<stdio.h>
#include<stdlib.h>
#include<string.h>

const int max=0x7ffffff;
int map[27][27],min,v[27],n,dis,m,sum,flag;
char cx,cy;

void Reset(int n)
{
	int i,j;
	map[n][n];
	memset(map,0,sizeof(map));
	for(i=0;i<n-1;i++)
	{
		scanf("%c%d%*c",&cx,&m);
//		cin>>cx>>m;
		for(j=0;j<m;j++)
		{
//			cin>>cy>>dis;
			scanf("%c%d%*c",&cy,&dis);
			map[cx-'A'][cy-'A']=map[cy-'A'][cx-'A']=dis;
		}
	}

	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
		{
			if(map[i][j]==0)
				map[i][j]=max;
		}
	}
}

int minTree(int n)
{
	int i,j;
	memset(v,0,sizeof(v));
	v[0]=1;
	sum=0;
	for(i=1;i<n;i++)
	{
		min=max;
		for(j=0;j<n;j++)
		{
			if(!v[j]&&map[0][j]<min)
			{
				min=map[0][j];
				flag=j;
			}
		}
		sum+=min;
		v[flag]=1;
		for(j=0;j<n;j++)
		{
			if(!v[j]&&map[0][j]>map[flag][j])
			{
				map[0][j]=map[flag][j];
			}
		}
	}
	return sum;
}

int main()
{
	while(scanf("%d%*c",&n),n)
	{
		Reset(n);
		printf("%d\n",minTree(n));
	}
	return 0;
}