POJ - 2031 Building a Space Station prim算法

/**
题意： 题目长的让人无语，就是简单的最小生成树模板题。
题目给出三维坐标系上的一些球的球心坐标和其半径，搭建通路，使得他们能够相互连通。
如果两个球有重叠的部分则算为已连通，无需再搭桥就是距离为0；
。求搭建通路的最小费用（费用就是边权，就是两个球面之间的距离）。
*/


#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;

const double inf=0x3f3f3f3f*1.0;
const int maxn=1e2+7;
double  dis[maxn],mmp[maxn][maxn],ans;
bool vis[maxn];
int n;

struct node
{
    double x,y,z,r;
}edge[maxn];

double far(node a,node b)
{
    return sqrt((double)((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)+(a.z-b.z)*(a.z-b.z)));
}

void build_map()
{
    for(int i=1;i<n;i++)
        for(int j=i+1;j<=n;j++)
    {
        double d=far(edge[i],edge[j]);
        double temp=edge[i].r+edge[j].r;
        if(d>temp) mmp[i][j]=mmp[j][i]=d-temp;
        else mmp[i][j]=mmp[j][i]=0;
    }
}

void prim()
{
    ans=0;
    for(int i=1;i<=n;i++)
    {
        dis[i]=mmp[1][i];
        vis[i]=0;
    }
    vis[1]=1;
    for(int i=1;i<=n;i++)
    {
        int k=-1;
        double Min=inf;///开始定义了int，这也是找了半天；
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&dis[j]<Min)
            {
                Min=dis[j];
                k=j;
            }
        }
        if(k==-1) break;
        vis[k]=1;
        ans+=Min;
        for(int j=1;j<=n;j++)
            if(!vis[j]&&dis[j]>mmp[k][j])
            dis[j]=mmp[k][j];
    }
    printf("%.3f\n",ans);
}

int main ()
{
    while(~scanf("%d",&n))
    {
        if(n==0) break;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            if(i==j) mmp[i][j]=0;
        else mmp[i][j]=inf;

        for(int i=1;i<=n;i++)
            scanf("%lf %lf %lf %lf",&edge[i].x,&edge[i].y,&edge[i].z,&edge[i].r);
            build_map();
            prim();
    }
    return 0;
}