HDU1875 畅通工程再续（最小生成树+并查集）

畅通工程再续

HERE

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!

知识点

最小生成树和并查集

思路

最小生成树板子题附加额外限制条件，Kruskal算法时用到到并查集。
只需要在选取最短边的时候外加判断即可。同时记录向出发点添加了多少个点，如果cnt != n - 1 则代表有的点没有进入树中，输出 oh! 。

代码

#include <iostream>
#include <algorithm>
#include <iomanip>
#include <cmath>
#define MAXN 105
#define INF 1500
using namespace std;
struct road{
	int u;
	int v;
	double w;
	bool operator<(road a)
	{
		return w<a.w?true:false;
	}
}edge[5000];
struct isa{
	int x;
	int y;
}node[MAXN];
int n=0,f[MAXN];
double map[MAXN][MAXN];
int find(int x)
{
	return x==f[x]?x:f[x]=find(f[x]);
}
int main()
{
	int t;
	cin>>t;
	while (t--)
	{
		cin>>n;
		for (int i=0;i<n;i++)
		{
			cin>>node[i].x>>node[i].y;
		}
		int cnt=0;
		for (int i=0;i<n-1;i++)
		{
			for (int j=i+1;j<n;j++)
			{
				edge[cnt].u=i;
				edge[cnt].v=j;
				edge[cnt++].w=sqrt((node[i].x-node[j].x)*(node[i].x-node[j].x)+(node[i].y-node[j].y)*(node[i].y-node[j].y));
			}
		}
		sort(edge,edge+n*(n-1)/2);
		cnt=0;
		double ans=0.0;
		for (int i=0;i<n;i++)
		{
			f[i]=i;
		}
		for (int i=0;i<n*(n-1)/2;i++)
		{
			if (edge[i].w<10||1000<edge[i].w) continue;
			int U=find(edge[i].u);
			int V=find(edge[i].v);
			if (U!=V)
			{
				f[U]=V;
				cnt++;
				ans+=edge[i].w;
			}
			if (cnt==n-1)
			{
				break;
			}
		}
		if (cnt==n-1)
		{
			cout<<fixed<<setprecision(1)<<ans*100<<endl;
		}
		else 
		{
			cout<<"oh!"<<endl;
		}
	}
	return 0;
}

tips：

太长时间没写都快忘了Kruskal是啥了，还是要及时写题解。