C - Building a Space Station - poj 2031

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<math.h>
#include<vector>
using namespace std;

#define maxn 105

struct point{double x, y, z, r;}p[maxn];
struct node
{
    int u, v;
    double len;

    friend bool operator < (node a, node b)
    {
        return a.len > b.len;
    }
};

int f[maxn];

double Len(point a, point b)//求两点间的距离
{
    double x = a.x - b.x;
    double y = a.y - b.y;
    double z = a.z - b.z;
    double l = sqrt(x*x+y*y+z*z)-a.r-b.r;

    if(l < 0)l = 0;

    return l;
}
int Find(int x)
{
    if(f[x] != x)
        f[x] = Find(f[x]);
    return f[x];
}

int main()
{
    int N;

    while(scanf("%d", &N) != EOF && N)
    {
        int i, j;
        node s;
        priority_queue<node>Q;

        for(i=1; i<=N; i++)
        {
            scanf("%lf%lf%lf%lf", &p[i].x, &p[i].y, &p[i].z, &p[i].r);
            f[i] = i;
        }

        for(i=1; i<=N; i++)
        for(j=i+1; j<=N; j++)
        {
            s.u = i, s.v = j;
            s.len = Len(p[i], p[j]);
            Q.push(s);
        }

        double ans = 0;

        while(Q.size())
        {
            s = Q.top();Q.pop();
            int u = Find(s.u), v = Find(s.v);

            if(u != v)
            {
                f[u] = v;
                ans += s.len;
            }
        }

        printf("%.3f\n", ans);
    }

    return 0;
}