ZOJ 1406 最小花費路

////////////////////////////////////////////////////////////
//最小花費路
//用最小產生樹Prim演算法
#include<iostream>
using namespace std;
#define MAX  1000
int c[28][28];

 

//先讓輸入轉化為鄰接矩陣儲存
void input(int n)
{
    int i,j,num,cost;
    char ch;
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            c[i][j]=MAX;
    for(i=1;i<=n;i++)
        c[i][i]=0;
    for(i=1;i<n;i++)
    {
        cin>>ch;
        cin>>num;
        while(num--)
        {
            cin>>ch>>cost;
            c[i][ch-'A'+1]=c[ch-'A'+1][i]=cost;
        }
    }
}

//這裡的closest[]也是沒有用到，他是用來記錄最小產生樹的路徑的
void prim(int n)
{
    int i,j,k,sum=0,tempmin;
    int *closest=new int[n];
    int *lowcost=new int[n];
    lowcost[1]=0;
    closest[1]=1;
    c[1][1]=1;
    for(i=2;i<=n;i++)
    {
        lowcost[i]=c[1][i];
        if(lowcost[i]!=MAX)
            closest[i]=1;
        else
            closest[i]=0;
    }
    for(i=2;i<=n;i++)
    {
        tempmin=MAX;
        for(j=1;j<=n;j++)
        {
            if(c[j][j]==0 && tempmin>lowcost[j])
            {
                tempmin=lowcost[j];
                k=j;
            }
        }
        sum+=tempmin;
        c[k][k]=1;
        for(j=1;j<=n;j++)
        {
            if(c[j][j]==0 && lowcost[j]>c[k][j])
            {
                lowcost[j]=c[k][j];
                closest[k]=j;
            }
        }
    }
    cout<<sum<<endl;
}

 

int main()
{
    int n;
    while(cin>>n && n!=0)
    {
        input(n);
        prim(n);
    }
    return 0;
}