hdu 1301 Jungle Roads

http://acm.hdu.edu.cn/showproblem.php?pid=1301

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 500
using namespace std;
const int inf=1<<30;

int g[maxn][maxn];
int dis[maxn];
bool vis[maxn];
int n,x,mm,sum;
char ch,ch1;

void prim()
{
    memset(vis,false,sizeof(vis));
    for(int i=0; i<n; i++) dis[i]=g[0][i];
    dis[0]=0;
    vis[0]=true;
    for(int i=1; i<n; i++)
    {
        int m=inf,x;
        for(int y=0; y<n; y++) if(!vis[y]&&dis[y]<m) m=dis[x=y];
        vis[x]=true;
        sum+=m;
        for(int y=0; y<n; y++) if(!vis[y]&&dis[y]>g[x][y]) dis[y]=g[x][y];
    }
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0) break;
        getchar();
        for(int i=0; i<27; i++)
        {
            for(int j=0; j<27; j++)
            {
                if(i==j) g[i][j]=0;
                else g[i][j]=inf;
            }
        }
        for(int i=1; i<=n-1; i++)
        {
            scanf("%c %d%*c",&ch,&x);
            for(int j=0; j<x; j++)
            {
                scanf("%c %d%*c",&ch1,&mm);
                g[ch-'A'][ch1-'A']=g[ch1-'A'][ch-'A']=mm;
            }
        }
        sum=0;
        prim();
        printf("%d
",sum);
    }
    return 0;
}