HDU1875--畅通工程再续

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!

 

/*
Kruskal算法
*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
using namespace std;
#define maxn 108
int father[maxn];
int find(int x)
{
	if(x == father[x]) return x;
	return father[x] = find(father[x]);
}
struct Point
{
	int x,y;
}point[maxn];
struct Edge
{
	int from,to;
	double len;
}edge[maxn*maxn];
bool cmp(Edge a,Edge b)
{
	return a.len < b.len;
}
double qlen(Point a,Point b)
{
	return sqrt(double((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)));
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d%d",&point[i].x,&point[i].y);
		}
		int k=1;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<i;j++)
			{
				double len=qlen(point[i],point[j]);
				if(len >= 10 && len <= 1000)
				{
					edge[k].from = i;
					edge[k].to = j;
					edge[k++].len = len;
				}
			}
		}
		for(int i=1;i<=n;i++)
		{
			father[i] = i;
		}
		sort(edge+1,edge+k,cmp);
		double cost = 0;
		int z = 0;
		for(int i=1;i<k;i++)
		{
			int uu = find(edge[i].from);
			int vv = find(edge[i].to);
			if(uu != vv)
			{
				cost += edge[i].len;
				father[uu] = vv;
				z++;
				if(z == n-1)break;
			}
		}
		if(z == n-1)
		{
			printf("%.1lf\n",cost*100);
		}
		else printf("oh!\n");
	}
	return 0;
}

优先队列优化的Prim

/*
边多用Prim。点多用Kruskal。
这题赶脚平均来看是边多。所以用Prim应该是更省时间的
*/
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
using namespace std;
#define inf 0x3f3f3f3f
#define maxn 108//点数
bool vis[maxn];
double dis[maxn];
int n;
struct Point 
{
	int x,y;
}point[maxn];
double qlen(Point a,Point b)
{
	return sqrt(double((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)));
}
struct Edge
{
	int v;
	double len;
	bool operator < (const Edge & a) const
	{
		return len > a.len;
	}
}edge;
vector <Edge> ans[maxn];
double heapprim()
{
	priority_queue <Edge> q;
	double cost = 0;
	edge.len = 0.;
	edge.v = 1;
	q.push(edge);
	int nown = 0;
	while(nown < n && !q.empty())
	{
L:
		edge = q.top();
		q.pop();
		while(vis[edge.v] && !q.empty()) goto L;
		if(!vis[edge.v])
		{
			int nowv = edge.v;
			nown++;
			vis[nowv] = 1;
			cost += edge.len;
			for(int i=0;i<ans[nowv].size();i++)
			{
				if(!vis[ans[nowv][i].v])
				{
					if(dis[ans[nowv][i].v] > ans[nowv][i].len)
					{
						dis[ans[nowv][i].v] = ans[nowv][i].len;
						edge.v = ans[nowv][i].v;
						edge.len = ans[nowv][i].len;
						q.push(edge);
					}
				}
			}
		}
	}
	if(nown < n) return -1;
	else return cost;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		memset(vis,0,sizeof(vis));
		for(int i=1;i<=n;i++)
		{
			dis[i] = inf;
		}
		for(int i=1;i<=n;i++)
		{
			ans[i].clear();
		}
		for(int i=1;i<=n;i++)
		{
			scanf("%d%d",&point[i].x,&point[i].y);
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<i;j++)
			{
				double len = qlen(point[i],point[j]);
				if(len >= 10 && len <= 1000)
				{
					edge.v = j;
					edge.len = len;
					ans[i].push_back(edge);
					edge.v = i;
					ans[j].push_back(edge);
				}
			}
		}
		double ans = heapprim();
		if(ans == -1)
		{
			printf("oh!\n");
		}
		else printf("%.1lf\n",ans*100);
	}
	return 0;
}


/*
接下来是不优化的Prim
*/
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
#define maxn 108
#define inf 0x3f3f3f3f
double W[maxn][maxn];
double dis[maxn];
bool vis[maxn];
struct Point 
{
	int x,y;
}point[maxn];
double qlen(Point a,Point b)
{
	return sqrt(double((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)));
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		memset(vis,0,sizeof(vis));
		memset(W,0,sizeof(W));
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d%d",&point[i].x,&point[i].y);
		}
		for(int i=1;i<=n;i++)
		{
			dis[i] = inf;
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<i;j++)
			{
				double len = qlen(point[i],point[j]);
				if(len >= 10 && len <= 1000)
				{
					W[i][j] = W[j][i] = len;
				}
			}
		}
		dis[1] = 0;
		bool flag = true;
		double cost = 0;
		for(int i=1;i<=n;i++)
		{
			double minlen = inf;
			int nowv;
			for(int j=1;j<=n;j++)
			{
				if(!vis[j])
				{
					if(minlen > dis[j])
					{
						minlen = dis[j];
						nowv = j;
					}
				}
			}
			if(minlen == inf && i!=n)
			{
				flag = false;
				break;
			}
			else 
			{
				vis[nowv] = 1;
				cost += minlen;
				for(int j=1;j<=n;j++)
				{
					if(!vis[j] && dis[j] > W[nowv][j] && W[nowv][j])
					{
						dis[j] = W[nowv][j];
					}
				}
			}
		}
		if(flag) printf("%.1lf\n",cost*100);
		else printf("oh!\n");
	}
	return 0;
}