poj 2031

题目概述

空间中有N个球，其坐标x,y,z，半径r，两个球在以下情况下视为相连，其一，两个球相交或相切，其二两个球表面由一条线段连接，其三两个球不相交，但都和另一个球相交，现在需要在球之间画若干线段使得所有球相连，求所画线段长度和最小值

时限

1000ms/3000ms

输入

第一行正整数N，其后N行，每行4个浮点数x,y,z,r，输入到N=0为止

限制

1<=N<=100;0< x,y,z,r<100

输出

每行一个保留三位小数的浮点数，为所求线段总长

样例输入

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

样例输出

20.000
0.000
73.834

讨论

图论，最小生成树，prim算法，一开始以为是计算几何，实际上也就是算个空间距离而已，思想就是算出两两球之间需要画线段的距离，然后一遍prim，最后对生成树求边长
从实现上，画线段的长度是球心距离扣掉两个球的半径，小于0的话就取0，除此之外就没有坑了

题解状态

268K，16MS，C++，1255B

题解代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define INF 0x3f3f3f3f
#define MAXN 103
#define memset0(a) memset(a,0,sizeof(a))
#define EPS 1e-16

int N;//球数
double x[MAXN], y[MAXN], z[MAXN], r[MAXN];//坐标以及半径
double graph[MAXN][MAXN], dist[MAXN];//图邻接矩阵以及生成树的边长 其实原来有个函数dis的 后来删了
bool mk[MAXN];//marked 标记节点是否已在树中
double fun()
{
    for (int p = 0; p < N; p++) {
        scanf("%lf%lf%lf%lf", &x[p], &y[p], &z[p], &r[p]);//input
        mk[p] = 0;//顺手清零
    }
    for (int p = 0; p < N; p++) {//下面初始化矩阵
        for (int i = 0; i < p; i++)
            graph[p][i] = graph[i][p];//主对角线下侧的元素直接照抄
        for (int i = p + 1; i < N; i++) {
            double d = sqrt((x[p] - x[i])*(x[p] - x[i]) + (y[p] - y[i])*(y[p] - y[i]) + (z[p] - z[i])*(z[p] - z[i])) - r[p] - r[i];//空间距离减两个半径
            graph[p][i] = d > 0 ? d : 0;//负数的话就是相交 相交就不用划线了
        }
    }
    double sum = 0;//下面是prim
    mk[0] = 1;
    for (int p = 0; p < N; p++)
        dist[p] = graph[0][p];
    for (int p = 1; p < N; p++) {
        double least = INF;
        int pos;
        for (int i = 0; i < N; i++)
            if (!mk[i] && dist[i] < least) {
                least = dist[i];
                pos = i;
            }
        mk[pos] = 1;
        sum += least;//在这里顺手求和
        for (int i = 0; i < N; i++)
            if (!mk[i] && dist[i]>graph[pos][i])
                dist[i] = graph[pos][i];
    }
    return sum;
}
int main(void)
{
    //freopen("vs_cin.txt", "r", stdin);
    //freopen("vs_cout.txt", "w", stdout);

    while (~scanf("%d", &N) && N)//input
        printf("%.3lf\n", fun());//output
}

EOF