1. 程式人生 > >【最小生成樹】Building a Space Station --Kruskal演算法+優先佇列+並查集

【最小生成樹】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;
}