POJ - 1251 Jungle Roads

题目链接：http://poj.org/problem?id=1251

 

Sample Input

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
3
A 2 B 10 C 40
B 1 C 20
0
Sample Output

216
30

 

分析：最小生成树模板题，注意将字母转化为数字ch-'A'+1，输入时%s%d，或者使用cin避免空格回车造成的影响

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<math.h>
#include<queue>
#include<map>
#include<time.h>
using namespace std;

#define INF 0x3f3f3f3f
#define N 153000

int maps[200][200],dist[50],vis[50],n;

void Init()
{
    for(int i=1; i<=26; i++)
    {
        for(int j=1; j<=26; j++)
        {
            if(i==j)
                maps[i][j]=0;
            else
                maps[i][j]=maps[j][i]=INF;
        }

        dist[i]=INF;
        vis[i]=0;
    }
}

int prim(int s)
{
    int i,j;

    vis[s]=1;
    for(i=1;i<=26;i++)
        dist[i]=maps[s][i];

    int sum=0;
    for(i=1;i<n;i++)
    {
        int Min=INF,index=-1;
        for(j=1;j<=26;j++)
        if(!vis[j]&&Min>dist[j])
        {
            Min=dist[j];
            index=j;
        }
        sum+=Min;
        vis[index]=1;

        for(j=1;j<=26;j++)
            if(!vis[j]&&dist[j]>maps[index][j])
            dist[j]=maps[index][j];

    }
    return sum;
}

int main()
{
    int i,j,m,c,h1,h2;
    char ch1,ch2;

    while(scanf("%d", &n), n)
    {
        Init();
        for(i=1; i<n; i++)
        {
            cin>>ch1>>m;
            h1=ch1-'A'+1;
            for(j=1; j<=m; j++)
            {
                cin>>ch2>>c;
                h2=ch2-'A'+1;
                maps[h1][h2]=maps[h2][h1]=c;
                ///printf("%d %d %d\n", h1,h2,maps[h1][h2]);
            }
        }

        int ans=prim(1);

        printf("%d\n", ans);
    }
    return 0;
}