POJ1251 MST

水题，难点在输入的处理。

#include<iostream>

#include<fstream>
using namespace std;
int n;
const int N=30;
int mat[N][N];
int dis[N];
const int inf=99999999;
bool visit[N];
void prim()
{
memset(visit,false,sizeof(visit));
for(int i=1;i<=n;i++)
dis[i]=mat[1][i];
visit[1]=true;
int now=1;
int ans=0;
for(int i=1;i<=n-1;i++)
{
int min=inf,mink=0;
for(int j=1;j<=n;j++)
{
if(!visit[j])
{
if(dis[j]>mat[now][j])
dis[j]=mat[now][j];
if(dis[j]<min)
min=dis[j],mink=j;
}
}
ans+=min;
now=mink;
visit[now]=true;
}
printf("%d\n",ans);
}
int main()
{
//freopen("C:\\Users\\wuyanyisb\\Desktop\\1.txt","r",stdin);
//fstream cin("C:\\Users\\wuyanyisb\\Desktop\\1.txt");
while(scanf("%d",&n),n!=0)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
mat[i][j]=inf;
char u,v;int tmp,w;
for(int i=1;i<=n-1;i++)
{
scanf(" %c%d",&u,&tmp);
for(int j=1;j<=tmp;j++)
{
scanf(" %c%d",&v,&w);

mat[u-'A'+1][v-'A'+1]=w;
mat[v-'A'+1][u-'A'+1]=w;
}
}
prim();
}
return 0;
}