Problem Description
相信大家都听说一个“百岛湖”的地方吧,百岛湖的居民生活在不同的小岛中,当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖,发展首先要解决的问题当然是交通问题,政府决定实现百岛湖的全畅通!经过考察小组RPRush对百岛湖的情况充分了解后,决定在符合条件的小岛间建上桥,所谓符合条件,就是2个小岛之间的距离不能小于10米,也不能大于1000米。当然,为了节省资金,只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。
Input
输入包括多组数据。输入首先包括一个整数T(T <= 200),代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数,接下来是C组坐标,代表每个小岛的坐标,这些坐标都是 0 <= x, y <= 1000的整数。
每组数据首先是一个整数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; }