#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<iostream>
#include<string>
using namespace std;
const int maxn=111111;
double x[111],y[111],z[111],r[111];
int fa[111];
int n,k;
struct node
{
int u,v;
double w;
bool operator<(const node &a)const
{
return w<a.w;
}
}p[maxn];
int find_fa(int x)
{
return (fa[x]==x)?fa[x]:fa[x]=find_fa(fa[x]);
}
double get_dis(double x1,double y1,double z1,double x2,double y2,double z2)
{
return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)+(z1-z2)*(z1-z2));
}
double kruskal()
{
double ans=0;
int cnt=0;
sort(p,p+k);
for(int i=0;i<k;i++)
{
int fx=find_fa(p[i].u);
int fy=find_fa(p[i].v);
if(fx!=fy)
{
fa[fx]=fy;
ans+=p[i].w;
cnt++;
}
if(cnt==n-1)
break;
}
return ans;
}
int main()
{
while(scanf("%d",&n)&&n)
{
for(int i=0;i<n;i++)
{
scanf("%lf%lf%lf%lf",&x[i],&y[i],&z[i],&r[i]);
fa[i]=i;
}
k=0;
double dist;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
dist=get_dis(x[i],y[i],z[i],x[j],y[j],z[j])-r[i]-r[j];
if(dist<0)
dist=0;
p[k].u=i;
p[k].v=j;
p[k++].w=dist;
}
}
double res=kruskal();
printf("%.3f\n",res);
}
return 0;
}
POJ2031(最小生成樹)
題目大意:給你n個球的座標以及半徑大小,求MST。如果兩球相切或相交,那麼間距爲0;如果兩球相離,那麼間距爲球心距減去半徑。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.