题目大意:太空中有一些球形的舱室cell,为了使宇航员们能够在舱室之间来去自如,你需要在必要的舱室之间搭建通道,通道的成本与长度有关,所以你必须选择最优的方案使搭建的通道长度总和最短。求这个最短总和长度。
分析:其实就是简单的最小生成树题,将cell看成节点,节点之间的边权是它们之间的距离(球面之间的距离,即球心距减去两个球的半径,若两个球有重叠,则边权应看做0)。
代码:
#include <iostream>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include <cstring>
using namespace std;
const double eps=1e-7;
const int MAXN=102;
const double INF=102.0;
class Cell{
public:
void getData(istream& in){
in>>x>>y>>z>>r;
}
double calDistance(Cell& cell){
double dx=cell.x-x,dy=cell.y-y,dz=cell.z-z;
double dis=sqrt(dx*dx+dy*dy+dz*dz)-r-cell.r;
return dis<0+eps?0:dis;
}
private:
double x,y,z,r;
};
Cell cell[MAXN];
double cost[MAXN][MAXN];
int n;
bool vis[MAXN];
void prim(){
memset(vis,false,sizeof vis);
int s=0;
vis[s]=true;
double sum_cost=0.0;
for(int i=1;i<n;++i){
double min_cost=INF;
int pos;
for(int j=1;j<n;++j){
if(!vis[j]&&cost[s][j]<min_cost){
min_cost=cost[s][pos=j];
}
}
vis[pos]=true;
sum_cost+=cost[s][pos];
for(int j=1;j<n;++j){
if(!vis[j]&&cost[s][j]>cost[pos][j]){
cost[s][j]=cost[pos][j];
}
}
}
cout<<fixed<<setprecision(3)<<sum_cost<<endl;
}
int main(){
while(cin>>n,n){
for(int i=0;i<n;++i){
cell[i].getData(cin);
}
for(int i=0;i<n;++i){
for(int j=0;j<n;++j){
if(i!=j){
cost[i][j]=cost[j][i]=cell[i].calDistance(cell[j]);
}
}
}
prim();
}
return 0;
}