Hdu 1301 Jungle Roads （最小生成树）

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

 

很明显，这是一道“赤裸裸”的最小生成树的问题；

我这里采用了Kruskal算法，当然用Prim算法也一样可以解题。

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
using namespace std;

typedef struct node{
    int from, to;
    int length;  //长度，指代问题中两地间的费用
}node;

const int maxn = 90;
node edge[maxn];
int pre[30];
int N,turn;

int cmp(const void *a, const void *b) // 比较函数
{
    return (*(node*)a).length-(*(node*)b).length;
}

inline int root( int i ) //用于判断是否两节点在同一树上
{
    while( pre[i] )
        i=pre[i];
    return i;
}

inline void input()  //输入
{
    char c1,c2;
    int i,order;
    int dis;
    turn=0;
    while(cin>>c1>>order)
    {
        for(i=0;i<order;i++)
        {
            cin>>c2>>dis;
            edge[turn].from=c1-'A';
            edge[turn].to=c2-'A';
            edge[turn].length=dis;
            turn++;
        }
        if( c1=='A'+N-2 ) //输入完成，退出
            break;
    }
    return ;
}

inline int span()
{
    memset(pre,0,sizeof(pre));
    qsort(edge,turn,sizeof(node),cmp); //根据长度排序
    int cost=0;
    int count=0;
    int pos=-1;
    int m,n;
    while(count<N-1)
    {
        do{
            pos++;
            m=root( edge[pos].from );
            n=root( edge[pos].to );
        }while(n==m);                 //用于判断是否两节点在同一树上
        cost = cost + edge[pos].length;
        pre[m]=n;             //把该节点加入这棵树
        count++;
    }
    return cost;
}

int main()
{
    while(cin>>N && N)
    {
        input();
        int flag=span();
        cout<<flag<<endl;
    }
    return 0;
}