POJ T2031 Building a Space Station
题解:
建造空间站之间的通道,空间站是球形,给你三维的坐标和半径,求最短的修建里程。
思路跟二维的没区别,且数据范围小。先求出每个点到其他点的距离。题目中说,如果两个空间站圆心之间距离小于或等于它们的半径之和,是不需要建通道的,所以可以判断下,并赋值为零。
而且最为重要的一点:通道长度是两个球体表面的距离(也就是圆心距离减去两半径之和)
其他的注意下输出格式就OK了 。
代码
#include<cstdio>
#include<iostream>
#include<cmath>
#define maxn 105
#define INF 0x3f3f3f3f
using namespace std;
int n;
double dis[maxn],sum;
double maps[maxn][maxn];
bool vis[maxn];
struct Node{
double x,y,z,r;
}node[maxn];
void init(){
sum = 0;
for(int i = 1; i <= n ; i++){
dis[i] = INF;
vis[i] = false;
/*for(int j = 1; j <= i; j++)
i == j ? maps[i][j] = 0 : maps[i][j] = maps[j][i] = INF;*/
}
}
double solve(int i,int j){
double a = (node[i].x - node[j].x)*(node[i].x - node[j].x);
double b = (node[i].y - node[j].y)*(node[i].y - node[j].y);
double c = (node[i].z - node[j].z)*(node[i].z - node[j].z);
return sqrt(a+b+c);
}
void Prim(int s){
dis[s] = 0;
for(int i = 1; i <= n ; i++){
int p = 0;
double minn = INF;
for(int j = 1; j <= n; j++){
if(!vis[j] && minn > dis[j]){
p = j;
minn = dis[j];
}
}
vis[p] = true;
sum += dis[p];
for(int j = 1; j <= n ; j++){
if(!vis[j] && dis[j] > maps[p][j])
dis[j] = maps[p][j];
}
}
}
int main(){
while(scanf("%d",&n),n){
init();
for(int i = 1; i <= n; i++)
scanf("%lf%lf%lf%lf",&node[i].x,&node[i].y,&node[i].z,&node[i].r);
for(int i = 1; i <= n ;i++)
for(int j = 1; j <= i ;j++){
if(i == j)
maps[i][j] = 0;
else{
double temp = solve(i,j);
if(temp > node[i].r+node[j].r)
maps[j][i] = maps[i][j] = temp - (node[i].r+node[j].r); //重点
else maps[j][i] = maps[i][j] = 0;
}
}
Prim(1);
printf("%.3lf\n",sum);
}
return 0;
}