最小生成樹 hdu 1233

還是暢通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 51181    Accepted Submission(s): 23313

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

Hint

Hint

Huge input, scanf is recommended.

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,m;
int map[5000];
int pre[5000];
struct node{
	int a,b;
	int len;
	bool operator < (const node & a) const{
		return len < a.len;	
	}
}s[5000];
void  init()// 初始化
{
	for(int i = 0; i < n*n; i++)
	{
		pre[i] = i;
	}
	return;
}

int Find(int a)//查找
{
	if(a != pre[a]) return pre[a] = Find(pre[a]);
	return a; 
}

int Merge(int x, int y)//連接
{
	
	int X = Find(x);
	int Y = Find(y);
	if(X!=Y) 
	{
		pre[X] = Y;
		return 1;
	}
		
	return 0;
}
int kruskal()
{
	sort(s,s+m);//排序
	int ans = 0;
	for(int i = 0 ; i < m; i++)
	{
		if(Find(s[i].a)!=Find(s[i].b))//如果沒有連接
		{
			ans += s[i].len;//加上最小路徑
			Merge(s[i].a , s[i].b);//並且連上
		 } 
	}
	return ans;//返回最小路徑
}
int main()
{
	while(scanf("%d",&n),n)
	{
		init();
		m = n*(n-1)/2;
		for(int i = 0; i < m; i++)
		{
			scanf("%d%d%d",&s[i].a,&s[i].b,&s[i].len);
		}
		int ans = kruskal();
		printf("%d\n",ans);
	}
}