1. 程式人生 > >hdu1233:還是暢通工程

hdu1233:還是暢通工程

http://acm.hdu.edu.cn/showproblem.php?pid=1233

Problem Description

某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可),並要求鋪設的公路總長度為最小。請計算最小的公路總長度。

 

 

Input

測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( < 100 );隨後的N(N-1)/2行對應村莊間的距離,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間的距離。為簡單起見,村莊從1到N編號。
當N為0時,輸入結束,該用例不被處理。

 

 

Output

對每個測試用例,在1行裡輸出最小的公路總長度。

 

 

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output

3
5

 

最小生成樹

#include<stdio.h>
#define N 120
int e[N][N],book[N],dis[N];
int main()
{
    int i,j,a,b,c,u,n,min,count,sum,inf=99999999;
    while(scanf("%d",&n),n!=0)
    {
        count=0;
        sum=0;
        for(i=1;i<=n;i++)
            for(j=i+1;j<=n;j++)
            {
                scanf("%d%d%d",&a,&b,&c);
                e[i][j]=e[j][i]=c;
            }

        for(i=1;i<=n;i++)
            book[i]=0;
        book[1]=1;
        for(i=1;i<=n;i++)
            dis[i]=e[1][i];
        count++;
        while(count<n)
        {
            min=inf;
            for(i=1;i<=n;i++)
                if(book[i]==0&&dis[i]<min)
                {
                    min=dis[i];
                    u=i;
                }
            book[u]=1;
            sum+=dis[u];
            count++;
            for(j=1;j<=n;j++)
                if(book[j]==0&&dis[j]>e[u][j])
                    dis[j]=e[u][j];
        }
        printf("%d\n",sum);
    }
    return 0;
}