HDU 1875畅通工程再续

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!

裸的最小生成树问题。注意到输入格式是以点的方式：各个点的坐标需要自己计算边权，那么采用普林算法更合适一些；而如果输入格式为边的方式类似（a b 10;a c 15.....)那么就采用克鲁斯卡尔算法。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#define inf 0x3f3f3f3f
using namespace std;
double map[101][101], box[101];
int flag[101], t, c;
double prim()
{
	int now = 1; 
	double res = 0;
	for (int i = 1;i <= c; i++)         //初始化box数组
		box[i] = map[now][i];
	for (int i = 2; i <= c; i++)
	{
		double minn = inf;
		for (int j = 2; j <= c; j++)
		{
			if (!flag[j] && box[j] < minn)
			{
				minn = box[j];
				now = j;
			}
		}
		if (minn != inf)               //由题意知会存在不能成边的情况，不能成边即找不到minn
		{
			flag[now] = 1;
			res += minn;
		}
		for (int j = 2; j <= c; j++)   //更新box数组
		{
			if (!flag[j]&&map[now][j] < box[j])
				box[j] = map[now][j];
		}
	}
	return res;
}
bool judge()                          //判断是否能形成mst，如果所有点都被标记则证明所有点都被选择过可成最小生成树，否则不能形成
{
	for (int i = 2; i <= c; i++)
	{
		if (flag[i] == 0)
			return false;
	}
	return true;
}
int main()
{
	scanf("%d", &t);
	while (t--)
	{
		memset(flag, 0, sizeof(flag));
		scanf("%d", &c);
		pair<double, double> p[101];      //常用pair容器来存储一对坐标
		for (int i = 1; i <= c; i++)
			cin >> p[i].first >> p[i].second;
		for (int i = 1; i <= c; i++)
		{
			for (int j = 1; j <= c; j++)
			{
					map[i][j] = sqrt(pow((p[i].first - p[j].first), 2) + pow((p[i].second - p[j].second), 2));
					if (((map[i][j] < 10) || (map[i][j]>1000))&&map[i][j]!=0)
						map[i][j] = inf;    //不能成边，则边权为inf
			}
		}
			double res = prim()*100;
			if(judge())
			printf("%.1f\n", res);
		    else
			printf("oh!\n");
	}
	return 0;
}