【最小生成树】Building a Space Station --Kruskal算法+优先队列+并查集

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
#define maxn 1005
//边的两点和其距离
struct node{
    int u,v;
    double len;
    //只有<重载操作符函数
    friend bool operator <(node a,node b)
    {
        return a.len>b.len;        //<为从大到小排列，>为从小到大排列
    }
};
//点的坐标和半径
struct Id{
    double x,y,z,r;    
}id[6000];
    
int n,m;
double ans;
double fa[maxn];
int par[maxn];
//并查集找老大
int find(int x){
    if(par[x]==x) return x;
    else{
        return par[x]=find(par[x]);
    }
}

double le(Id n,Id m){
    double a=(n.x-m.x);
    double b=(n.y-m.y);
    double c=(n.z-m.z);
    double mm=sqrt(a*a+b*b+c*c)-n.r-m.r;
    //a,b,c可能为0会导致mm为负数，必须判断！！
    if(mm>0){
        return mm;
    }
    else{
        return 0;
    }
}

int main(){
    while(scanf("%d",&n),n){
        node s;
        priority_queue<node> qq;//从大到小
        
        for(int i=1;i<=n;i++){
            scanf("%lf %lf %lf %lf",&id[i].x,&id[i].y,&id[i].z,&id[i].r);
            par[i]=i;//初始化，每个点就是一个集合
        }
        //循环得到任意两点距离
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){    
                    s.u=i;
                    s.v=j;
                    s.len=le(id[i],id[j]);
                    qq.push(s); //加进队列中 （虚边还未连接）
                }    
            }
        double ans=0;
    while(!qq.empty()){
        node w=qq.top();//取队列头的一条虚边
        qq.pop();
        //边的两个点是否是一个集合（是的话再连接就成回路了）
        int a=find(w.u);
        int b=find(w.v);
        if(a!=b){
            par[a]=b;
            ans+=w.len;
        }
    }
        printf("%.3f\n",ans);
    
    }
    return 0;
}