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