POJ Building a Space Station

题目链接

题目大意：太空中有一些球形的舱室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;
}