HDU 1875 暢通工程再續 Kruskal

暢通工程再續

http://acm.hdu.edu.cn/showproblem.php?pid=1875
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Problem Description

相信大家都聽說一個“百島湖”的地方吧，百島湖的居民生活在不同的小島中，當他們想去其他的小島時都要通過劃小船來實現。現在政府決定大力發展百島湖，發展首先要解決的問題當然是交通問題，政府決定實現百島湖的全暢通！經過考察小組RPRush對百島湖的情況充分了解後，決定在符合條件的小島間建上橋，所謂符合條件，就是2個小島之間的距離不能小於10米，也不能大於1000米。當然，爲了節省資金，只要求實現任意2個小島之間有路通即可。其中橋的價格爲 100元/米。

Input

輸入包括多組數據。輸入首先包括一個整數T(T <= 200)，代表有T組數據。
每組數據首先是一個整數C(C <= 100),代表小島的個數，接下來是C組座標，代表每個小島的座標，這些座標都是 0 <= x, y <= 1000的整數。

Output

每組輸入數據輸出一行，代表建橋的最小花費，結果保留一位小數。如果無法實現工程以達到全部暢通，輸出”oh!”.

Sample Input

2
2
10 10
20 20
3
1 1
2 2
1000 1000

Sample Output

1414.2
oh!

Source

2008浙大研究生複試熱身賽（2）——全真模擬

C++

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

using namespace std;
const int MAXN = 100 + 10;
const int MAXM = 10000 + 10;
int fa[MAXN];
int n, m;

struct node{
	int x, y;
	double cost;
}point[MAXN], edge[MAXM];

bool cmp(node a, node b){
	return a.cost < b.cost;
}

void init()
{
	for(int i = 1; i <= m; i ++)
		fa[i] = i;
}

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

void kruskal()
{
	init();
	int cnt = 0;
	double ans = 0;
	sort(edge, edge + n, cmp);
	for(int i = 0; i < n; i ++)
	{
		int tx = find(edge[i].x);
		int ty = find(edge[i].y);
		if(tx != ty)
		{
			fa[tx] = ty;
			cnt ++;
			ans += edge[i].cost;
		}
		if(cnt == m - 1)
			break;
	}
	if(cnt < m - 1) printf("oh!\n");
	else printf("%.1lf\n", ans * 100);
}

int main()
{
	int T;
	scanf("%d", &T);
	while(T --)
	{
		scanf("%d", &m);
		n = 0;
		double tmp;
		for(int i = 0; i < m; i ++)
		{
			scanf("%d%d", &point[i].x, &point[i].y);
			for(int j = 0; j < i; j ++)
			{
				tmp = sqrt((point[i].x-point[j].x)*(point[i].x-point[j].x)+(point[i].y-point[j].y)*(point[i].y-point[j].y)); 
				if(tmp >= 10 && tmp <= 1000)
				{
					edge[n].x = i + 1;
					edge[n].y = j + 1;
					edge[n ++].cost = tmp;
				}
			}
		}
		kruskal();
	}
	return 0;
}