POJ 2031 Building a Space Station(kuangbin带你飞 专题六：最小生成树)

大概题意就是要有n个空间站，给出三维坐标以及半径（是个球）..要每两个空间站之间都能互相通信啥的...问建立的最小花费是多少..花费就是两个空间站的距离...如果小于0当做0..

两球之间的距离等于两球心之间的距离减去两球的半径...就能算出球面之间的距离了..然后最小生成树走起

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
double x,y,z,r;
};
struct edge
{
int x,y;
double s;
};
int cmp(edge a,edge b)
{
return a.s<b.s;
}
int f[150],n;
void inti()
{
for(int i=1;i<=n;i++)
f[i]=i;
}
int find(int x)
{
if(x!=f[x])
return f[x]=find(f[x]);
return x;
}
double dist(node A,node B)
{
return sqrt((A.x-B.x)*(A.x-B.x)+(A.y-B.y)*(A.y-B.y)+(A.z-B.z)*(A.z-B.z))-A.r-B.r;
}
int main()
{
node p[120];
edge s[12000];
while(cin>>n,n)
{
inti();
for(int i=1;i<=n;i++)
cin>>p[i].x>>p[i].y>>p[i].z>>p[i].r;
int k=0;
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
s[k].x=i,s[k].y=j;
s[k].s=dist(p[i],p[j]);
s[k].s=max(s[k].s,0.0);
k++;
}
}
sort(s,s+k,cmp);
int t=0;
double ans=0;
for(int i=0;i<k;i++)
{
int x=find(s[i].x);
int y=find(s[i].y);
if(x!=y)
{
f[x]=y;
ans+=s[i].s;
t++;
}
if(t==n-1)
break;
}
printf("%.3f\n",ans);
}
return 0;
}