POJ 2031 Building a Space Station

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

题目意思是给出三维坐标系上的一些球的球心坐标和其半径，搭建通路，使得他们能够相互连通。如果两个球有重叠的部分则算为已连通，无需再搭桥。求搭建通路的最小费用（费用就是边权，就是两个球面之间的距离）。就是简单的最小生成树。

prim：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
const int MAXN = 105;
struct data {
    double x , y , z , r;
}a[MAXN];
double d[MAXN] , cost[MAXN][MAXN] , INF = 1e9;
bool vis[MAXN];

void init(int n) {
    for(int i = 0 ; i <= n ; i++) {
        for(int j = 0 ; j <= n ; j++)
            cost[i][j] = INF;
        d[i] = INF;
        vis[i] = false;
    }
}

inline double f(double x , double y) {
    return (x - y) * (x - y);
}

double dis(int i , int j) {
    double temp = f(a[i].x , a[j].x) + f(a[i].y , a[j].y) + f(a[i].z , a[j].z);
    return sqrt(temp);
}

double prim(int s , int n) {
    d[s] = 0;
    double res = 0;
    while(1) {
        int u = -1;
        double Min = INF;
        for(int i = 1 ; i <= n ; i++) {
            if(!vis[i] && Min > d[i]) {
                Min = d[i];
                u = i;
            }
        }
        if(u == -1)
            return res;
        vis[u] = true;
        res += d[u];
        for(int i = 1 ; i <= n ; i++) {
            d[i] = (d[i] > cost[u][i] ? cost[u][i] : d[i]);
        }
    }
}

int main()
{
    int n;
    ios::sync_with_stdio(false);
    while(cin >> n && n) {
        init(n);
        for(int i = 1 ; i <= n ; i++) {
            cin >> a[i].x >> a[i].y >> a[i].z >> a[i].r;
        }
        for(int i = 1 ; i < n ; i++) {
            for(int j = i + 1 ; j <= n ; j++) {
                double temp = a[i].r + a[j].r , temp2 = dis(i , j);
                if(temp >= temp2)
                    cost[i][j] = cost[j][i] = 0;
                else
                    cost[i][j] = cost[j][i] = temp2 - temp;
            }
        }
        double res = prim(1 , n);
        printf("%.3f
" , res);
    }
}

kruskal:就是把边从小到大排序一下，用并查集判断祖节点是否相同，不同就合并。