Hdu 1301 - Jungle Roads

最小生成树 - Prim算法

 

 

 

题意：n为村庄数目（2 < n < 27），接下来的n-1行里，先输入一个字母，代表从这个点开始出发；接着是一个数字k，k代表从前面那个字母所代表的村庄 到 后面将会输入的字母所代表的村庄的路数，之后再依次输入k组数据，其中的字母后的数字代表这一行里第一个字母到该字母的那条路的费用。当k = 0 时，说明从此点出发，不到达其他村庄。费用是不大于100的正整数。

 

例如：A 2 B 12 I 25

代表：从 A 出发一共能到达两个村庄，分别是 B 和 I ，A 到 B 的路费是12，A 到 I 的路费是25

 

 

第一次WA了，原因是 map[ ][ ] 的初始化代码写的是

memset(map, max, sizeof(map));

 

然后改成以下初始化代码就AC了

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

 

AC代码：

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

const int max = 0x7ffffff;
int map[105][105], low[105], v[105];
int n;

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

int main()
{
    char s1,s2;
    int m,c,i,j,a,b;
    while(scanf("%d",&n),n)
    {
        //getchar();
        //从点1到点n的初始化。注意：不能用memset(map, max, sizeof(map))代替！
        //从点1开始，因为我是从点1开始贪心的。
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
            {
                map[i][j] = max;
            }
        for(i=1; i<n; i++)
        {
            getchar();
            scanf("%c%d",&s1,&m);
            for(j=1; j<=m; j++)
            {
                getchar();
                scanf("%c%d",&s2,&c);
                a = s1-'A'+1;   //重点：因为从点1开始贪心，所以这里是+1；
                b = s2-'A'+1;   //同理：如果从点0开始贪心，就是s1(s2)-'A';
                map[b][a] = map[a][b] = c;
            }
        }
        printf("%d\n",MinTree(n));
    }
    return 0;
}