oh!
题目大意:给你几个岛的坐标,只有两个岛的距离在10~1000米范围内的岛才能
建立道路。每米道路花费100元。问:能否建立连接全部岛屿的道路。若能,输出
建立道路的最小花费;若不能输出"oh!"。
思路:主要是建图问题,纠结在Prim上半天,最后才发现只要建图的时候多想想就
好了。建图的时候,只有满足条件的才能赋值距离,否则就为INF(假设的无穷大)。
Prim求最小生成树的时候,若没有找到与当前链接岛屿最近的满足建造条件的道路,
就直接输出"oh!",并退出。否则就继续查找,直到最后输出最小花费。
#include<iostream> #include<algorithm> #include<cstdio> #include<cstring> #include<cmath> using namespace std; struct City { int x; int y; }A[110]; double G[110][110],low[110]; int vis[110]; void Prim(int N) { memset(vis,0,sizeof(vis)); vis[1] = 1; int pos = 1; double ans = 0; for(int i = 1; i <= N; i++)//先求点1到各点的距离 { if(i != pos) low[i] = G[pos][i]; } int count = 1; for(int i = 1; i < N; i++) { double Min = 100000000000; for(int j = 1; j <= N; j++)//找到距离点pos最近的点和距离 { if(!vis[j] && Min > low[j]) { Min = low[j]; pos = j; } } if(Min < 100000000000) //找到了 { ans += Min; vis[pos] = 1; } else //不存在 { printf("oh!\n"); return; } for(int j = 1; j <= N; j++) //更新pos点到各点的距离看是否找到离遍历过的点更近的距离。 { if(!vis[j] && low[j] > G[pos][j]) { low[j] = G[pos][j]; } } } printf("%.1lf\n",ans*100); } int main() { int T,N; double Dist,x,y; cin >> T; while(T--) { cin >> N; memset(A,0,sizeof(A)); for(int i = 1; i <= N; i++)//初始化图G for(int j = 1; j <= N; j++) G[i][j] = 100000000000; for(int i = 1; i <= N; i++) { cin >> A[i].x >> A[i].y; } for(int i = 1; i <= N; i++) { for(int j = i; j <= N; j++) { if(i == j)//自身节点i到i长度为0 { G[i][j] = 0.0; continue; } x = A[i].x-A[j].x; y = A[i].y-A[j].y; Dist = sqrt(x*x + y*y); if(Dist >= 10.0 && Dist <= 1000.0)//满足条件才有长度,否则无限长 G[i][j] = G[j][i] = Dist; } } Prim(N); } return 0; }