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;
}