C - Building a Space Station - poj 2031

空间站是有一些球状的房间组成的，现在有一些房间但是没有相互连接，你需要设计一些走廊使他们都相通，当然，有些房间可能会有重合（很神奇的样子，重合距离是0），你需要设计出来最短的走廊使所有的点都连接。

分析：因为给的都是点的坐标，所以构图的时候会有一些麻烦，不过也仅此而已。。。

*******************************************************************

 

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