POJ 2031 Building a Space Station Kruskal最小生成树

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/C

题意：大意是，给定一些空间站，把空间站当作球来处理，然后给定每个空间站的描述，球心坐标和半径，求两两连通的最小代价。两个求相连通的条件是：

1：相互相连或是重叠（直接相连）

2：通过第三者相连（间接相连）

思路：把直接相连的用并查集预处理就行了。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define maxn 110

using namespace std;
struct edge{
    int u,v;
    double w;
    bool operator<(const edge &a)const{
        return w<a.w;
    }
}E[maxn*maxn];

double X[maxn],Y[maxn],Z[maxn],R[maxn];
int N,tot,pre[maxn];

int find(int x){
    int t=x;
    while(t!=pre[t]) t=pre[t];
    while(x!=t) pre[x]=t,x=pre[x];
    return t;
}

void Kruskal(){
    double sum=0;
    for(int i=0;i<tot;i++){
        int fu=find(E[i].u),fv=find(E[i].v);
        if(fu==fv) continue;
        sum+=E[i].w;pre[fv]=fu;
    }
    printf("%.3f\n",sum);
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    while(cin>>N && N){
        for(int i=1;i<=N;i++)
            scanf("%lf %lf %lf %lf",X+i,Y+i,Z+i,R+i);
        for(int i=0;i<=N;i++) pre[i]=i;
        tot=0;
        for(int i=1;i<=N;i++) for(int j=i+1;j<=N;j++){
            double t=sqrt((X[i]-X[j])*(X[i]-X[j])+(Y[i]-Y[j])*(Y[i]-Y[j])+(Z[i]-Z[j])*(Z[i]-Z[j]));
            if(R[i]+R[j]>=t) pre[find(j)]=find(i);
            else E[tot++]=(edge){i,j,t-R[i]-R[j]};//主要是得注意这里
        }
        sort(E,E+tot);
        Kruskal();
    }
    return 0;
}