hdu1233 還是暢通工程

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

Hint

Hint

Huge input, scanf is recommended.

AC Code

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <limits.h>
#include <queue>
#define inf 0x3f3f3f3f
using namespace std;




int main(int argc, char** argv) {
	int n,e[110][110],dis[110],book[110];
	while(cin>>n&&n){
		int i,j,sum=0;
		memset(book,0,sizeof(book));
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				if(i==j) e[i][j]=0;
				else e[i][j]=inf;
			}
		for(i=1;i<=n*(n-1)/2;i++)
		{
			int a,b,w;
			scanf("%d%d%d",&a,&b,&w);
			e[a][b]=e[b][a]=w;
		}
		for(i=1;i<=n;i++)
			dis[i]=e[1][i];
		book[1]=1;
		for(i=1;i<n;i++)
		{
			int min=inf,id;
			for(j=1;j<=n;j++)
			{
				if(book[j]==0&&dis[j]<min)
				{
					min=dis[j];
					id=j;
				}	
			} 
			book[id]=1;sum+=min;
			for(j=1;j<=n;j++)
			{
				if(book[j]==0&&dis[j]>e[id][j])
					dis[j]=e[id][j];
			}
		}
		cout<<sum<<endl;
	} 
	return 0;
}