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;
}