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