HDU - 1875 畅通工程再续

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组RPRush对百岛湖的情况充分了解后，决定在符合条件的小岛间建上桥，所谓符合条件，就是2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。

Input

输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数，接下来是C组坐标，代表每个小岛的坐标，这些坐标都是 0 <= x, y <= 1000的整数。

Output

每组输入数据输出一行，代表建桥的最小花费，结果保留一位小数。如果无法实现工程以达到全部畅通，输出”oh!”.

Sample Input

Sample Output
1414.2
oh!

解题思路：最短路径，可以使用kruskal算法来做，也可以使用prime算法来做，题目给出的是顶点的信息，所以要找到所有距离小于1000大于10的边，然后使用kruskai算法找最短路

AC代码：

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

using namespace std;
const int maxn=1e5;
double ans;

struct node{
	double x,y;
	int id;
}a[maxn];

struct edge{
	int u;
	int to;
	double len;
}e[maxn];

int pre[maxn],ran[maxn];

bool cmp(edge a,edge b)
{
	return a.len<b.len;
}
void init(int n)
{
	for(int i=0;i<n;i++)
	{
		pre[i]=i;
		ran[i]=0;
	}
} 

int find(int x)
{
	return x==pre[x]?x:(pre[x]=find(pre[x]));
}

void merge(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y)
		return ;
	if(ran[x]<ran[y])
		pre[x]=y;
	else
	{
		pre[y]=x;
		if(ran[x]==ran[y])
			ran[x]++;
	}
	
}

bool same(int x,int y)
{
	return find(x)==find(y);
}

int kruskal(int n,int m)//n为边数，m为顶点数 
{
	int nedge=0;
	for(int i=0;i<n&&nedge!=m-1;i++)
	{
		if(!same(e[i].u,e[i].to))
		{
			merge(e[i].u,e[i].to);
			ans+=e[i].len;
			nedge++;
		}
	}
	return nedge;
}
int main()
{
	int t,n;
	scanf("%d",&t);
	while(t--)
	{
		int cnt=0;
		ans=0;
		scanf("%d",&n);
		init(n);
		for(int i=0;i<n;i++)
		{
			scanf("%lf%lf",&a[i].x,&a[i].y);
			a[i].id=i;
		}
		for(int i=0;i<n-1;i++)//找边 
		{
			for(int j=i+1;j<n;j++)
			{
				double x1=a[j].x-a[i].x;
				double y1=a[j].y-a[i].y;
				if((x1*x1+y1*y1)>=100&&(x1*x1+y1*y1)<=1000000)
				{
					e[cnt].u=a[i].id;
					e[cnt].to=a[j].id;
					e[cnt++].len=sqrt(x1*x1+y1*y1);
				}
			}
		}
		sort(e,e+cnt,cmp);
		int term=kruskal(cnt,n);
		if(term!=n-1)//如果所有顶点联通，那么最小生成树的边数一定等于n-1 
			printf("oh!\n");
		else
			printf("%.1lf\n",ans*100);
	}
	return 0;
}

HDU - 1875 畅通工程再续

猜你喜欢