poj2301Building a Space Station（最小生成树）

题目链接：http://poj.org/problem?id=2031

求出任意两点之间的距离减去两个点的半径。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=110;
int n;
int f[maxn];
struct node
{
    double x,y,z;
    double r;
}p[maxn];

struct edge
{
    int u,v;
    double w;
    bool operator < (const edge &a)
    {
        return w<a.w;
    }
}e[100010];  //不知道要开多大！

double getdis(int  i,int j)
{
    double dx=p[i].x-p[j].x;
    double dy=p[i].y-p[j].y;
    double dz=p[i].z-p[j].z;
    return sqrt(dx*dx+dy*dy+dz*dz)-p[i].r-p[j].r;
}

void init()
{
    for(int i=0;i<n;i++)
        f[i]=i;
}

int gf(int a)
{
    return a==f[a]?a:f[a]=gf(f[a]);
}

void uni(int a,int b)
{
    int pa=gf(a);
    int pb=gf(b);
    f[pb]=pa;
}

int main()
{
    while(scanf("%d",&n)&&n)
    {
        init();
        for(int i=0;i<n;i++)
        {
            scanf("%lf%lf%lf%lf",&p[i].x,&p[i].y,&p[i].z,&p[i].r);
        }
        int cnt=0;
        for(int i=0;i<n;i++)
            for(int j=i+1;j<n;j++)
        {
            e[cnt].u=i;
            e[cnt].v=j;
            e[cnt].w=getdis(i,j);
            cnt++;
        }
        double ans=0;
        sort(e,e+cnt);
        for(int i=0;i<cnt;i++)
        {
            if(e[i].w<=0) uni(e[i].u,e[i].v);
            else   if(gf(e[i].u)!=gf(e[i].v))  {ans+=e[i].w; uni(e[i].u,e[i].v);}
        }
        printf("%.3f
",ans);
    }
}