最小生成樹--還是暢通工程

還是暢通工程

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

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


        
  
Huge input, scanf is recommended.

Hint

Hint
        
 

 

#include<stdio.h>
#include<algorithm>
#define MAX 1005
using namespace std; 

struct Edge{
	int u,v,len;//uv表示邊所連接的兩個頂點，len表示距離 
	const bool operator< (const Edge & rhs)const{//重載比較運算符 
		return len < rhs.len;//根據距離從小到大排序 
	}
}edge[6000];

int fa[MAX];
int n,m;//n個頂點，m條路 

int find(int x){
	if(fa[x]==x) return x;
	else return fa[x] = find(fa[x]);
}

int kruskal(){
	int ans = 0,cnt = 0;//ans表示答案，cnt表示找到邊的數組 
	for(int i = 0;i<=n;i++)//初始化 
		fa[i] = i;
	sort(edge,edge+m);//排序所有邊 
	for(int i = 0;i<m;i++){//開始對排序好的邊進行遍歷 
		int x = find(edge[i].u);
		int y = find(edge[i].v);
		if(x!=y){
			ans += edge[i].len;
			fa[x] = y;
			cnt ++;
		}
		if(cnt==n-1) break; //找完n-1條邊 
	}
	return ans;
}
int main(){
	while(scanf("%d",&n)!=EOF&&n!=0){
		//輸入 
		m = n*(n-1)/2;
		for(int i = 0;i<m;i++){
			scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].len);
		}
		//核心代碼 
		int len = kruskal();
		//輸出 
		printf("%d\n",len);
	}
	return 0;
}