poj2031 最小生成树

题目链接在这里

题目大意：

在三维空间中给出n个位置，每个位置有一个球，每个球有x, y, z ,r四个参数代表x、y、z轴坐标和半径r。球和球之间的连接在球的表面积上进行。求让它们相互连通所需要的最小的距离。

解题思路：

把每个球之间相连所需要的距离算出来（就是建边），然后求一棵最小生成树就好了。

 

注意要用double!!!!用float过不了，把float给define成double才过的。

注意double的输出是%f，不是%lf

代码如下：

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
#define float double
#define rep(i, x) for(int i = 0; i < x; ++i)
#define clr(x) memset(x, 0, sizeof(x))
using namespace std;
const int MaxN = 110;

struct Node{
    float x, y, z, r;
}node[MaxN];    //存储每个球的位置

struct Edge{;
    int u, v;
    float val;

    bool operator < (const Edge &e) const{
        return val > e.val;
    }
};    //球和球之间建立的边

int n;
int par[MaxN], r[MaxN];

float getDis(const Node& x, const Node& y){    //算球和球之间的距离
    float xx = (x.x - y.x) * (x.x - y.x);
    float yy = (x.y - y.y) * (x.y - y.y);
    float zz = (x.z - y.z) * (x.z - y.z);
    float ans = sqrt(xx + yy + zz) - x.r - y.r;
    if(ans < 0.0) ans = 0.0;    //如果减去半径之后距离为负，注意改成0
    return ans;
}

int Find(int x){
    if(x == par[x]) return x;
    return par[x] = Find(par[x]);
}

void unite(int x, int y){
    x = Find(x);
    y = Find(y);
    if(x == y)  return;
    if(r[x] < r[y]) par[x] = y;
    else{
        par[y] = x;
        if(r[x] == r[y])    ++r[x];
    }
}

bool check(int x, int y){
    return Find(x) == Find(y);
}



void solve(){
    priority_queue<Edge> que;
    float ans = 0.0;
    rep(i, n){
        for(int j = i + 1; j < n; ++j){
            que.push(Edge{i, j, getDis(node[i], node[j])});
        }
    }
    rep(i, MaxN) par[i] = i;
    clr(r);
    while(--n){
        Edge e = que.top();
        que.pop();
        if(check(e.u, e.v)){
            ++n;
            continue;
        }
        unite(e.u, e.v);
        ans += e.val;
    }
    printf("%.3f\n", ans);
}

int main(){
    ios::sync_with_stdio(false);
    while(cin >> n, n){
        rep(i, n)   cin >> node[i].x >> node[i].y >> node[i].z >> node[i].r;
        solve();
    }

    return 0;
}