POJ

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

Sample Input
3
10.000 10.000 50.000 10.000
40.000 10.000 50.000 10.000
40.000 40.000 50.000 10.000
2
30.000 30.000 30.000 20.000
40.000 40.000 40.000 20.000
5
5.729 15.143 3.996 25.837
6.013 14.372 4.818 10.671
80.115 63.292 84.477 15.120
64.095 80.924 70.029 14.881
39.472 85.116 71.369 5.553
0

Sample Output
20.000
0.000
73.834

最小生成树模板题

题意：

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

*：给的是点坐标，注意构图时，仍需要搭建的距离是 两圆心距减去两圆半径

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<math.h>
#include<queue>
using namespace std;
#define INF 0x3f3f3f3f
#define N 12342

int n;
int vis[N];
double Map[120][120],dist[N];

struct node
{
    double x,y,z,r;
}p[N];

double Len(node a,node 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;
}

void Init()
{
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=n;j++)
        Map[i][j]=(i==j)?0:INF;

        dist[i]=INF;
        vis[i]=0;
    }
}

double Prim(int s)
{
    vis[s]=1;

    for(int i=1;i<=n;i++)
        dist[i]=Map[s][i];

    double sum=0.0;

    for(int i=1;i<n;i++)
    {
       double Min=INF;
       int index=-1;

        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&Min>dist[j])
            {
                Min=dist[j];
               ///printf("%lf
", Min);
                index=j;
            }
        }
        sum+=Min;
        ///printf("%lf
", sum) ;

        vis[index]=1;

        for(int j=1;j<=n;j++)
            if(!vis[j]&&dist[j]>Map[index][j])
            dist[j]=Map[index][j];
    }
    return sum;
}

int main( )
{
    while(scanf("%d", &n), n)
    {
        Init();

        for(int i=1;i<=n;i++)
            scanf("%lf%lf%lf%lf", &p[i].x,&p[i].y,&p[i].z,&p[i].r);

        ///记录各细胞间还需搭建的距离
        for(int i=1;i<=n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                Map[i][j]=Map[j][i]=Len(p[i],p[j]);
               ///printf("%d %d %lf
", i,j,Map[i][j]);
            }
        }

        double ans=Prim(1);

        printf("%.3f
", ans);
    }
    return 0;
}

本人博客：http://www.cnblogs.com/weiyuan/