HDU 1301 Jungle Roads(Prim求最小生成树)

题目地址:点击打开链接

思路:最小生成树模板题,刚开始各条路之间的距离设为INT_MAX(int类型的最大数)

AC代码:

#include <iostream>
#include<cstring>
#include<climits>

using namespace std;

int map[30][30],visit[30],lowcost[30];
int main()
{
    char a,b;
    int n,i,j,num,num1,min,k,sum;
    while(cin>>n && n)
    {
        sum = 0;
        for(i=0; i < 30; i++)
        {
            for(j=0; j<30; j++)
                map[i][j] = INT_MAX;
        }
        memset(visit,0,sizeof(visit));
        for(i=1; i<n; i++)
        {
            cin>>a>>num;
            for(j=0; j<num; j++)
            {
                cin>>b>>num1;
                map[a-'A'][b-'A'] = num1;
                map[b-'A'][a-'A'] = num1;
            }
        }
        for(i=0; i<n; i++)
        {
            lowcost[i] = map[0][i];
        }
        visit[0] = 1;
        for(i=1; i<n; i++)
        {
            min = INT_MAX;
            for(j=0; j<n; j++)
            {
                if(visit[j] == 0 && lowcost[j] < min)
                {
                    min = lowcost[j];
                    k = j;
                }
            }
            visit[k] = 1;
            sum += min;
            for(j=0; j<n; j++)
            {
                if(visit[j] == 0 && map[k][j] < lowcost[j])
                    lowcost[j] = map[k][j];
            }
        }
        cout<<sum<<endl;
    }
    return 0;
}


你可能感兴趣的:(HDU 1301 Jungle Roads(Prim求最小生成树))