POJ2031（最小生成树）

题目大意：给你n个球的坐标以及半径大小，求MST。如果两球相切或相交，那么间距为0；如果两球相离，那么间距为球心距减去半径。

#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;
}