[kuangbin带你飞]专题六 最小生成树 C POJ 2031

题目地址：https://vjudge.net/contest/66965#problem/C

思路：最小生成树，模板题，距离即为两个球的圆心距减去半径。

AC代码：

#include<iostream>#include<cstdio>#include<algorithm>#include<cmath>using namespace std;const int maxn=100+10;struct pos{    double x,y,z;}E[maxn*maxn];int fa[maxn];struct point{    double x,y,z,r;}a[maxn];int n;double get(int i,int j){    return sqrt((a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y)+(a[i].z-a[j].z)*(a[i].z-a[j].z));}int find(int p){    if(p!=fa[p])        fa[p]=find(fa[p]);    return fa[p];}double cmp(pos a,pos b){    return a.z<b.z;}int main(){    while(scanf("%d",&n) && n)    {        for(int i=0;i<n;i++)        {            cin>>a[i].x>>a[i].y>>a[i].z>>a[i].r;        }        int m=0;        for(int i=0;i<n;i++)            for(int j=0;j<n;j++)        {            if(i==j)                continue;            double temp=get(i,j)-a[i].r-a[j].r;            if(temp<0)                temp=0;            E[m].x=i,E[m].y=j,E[m++].z=temp;        }        sort(E,E+m,cmp);        for(int i=0;i<n;i++)            fa[i]=i;        double ans=0;        for(int i=0;i<m;i++)        {        int x=E[i].x,y=E[i].y;        int fx=find(x);        int fy=find(y);        if(fx!=fy)        {            fa[fx]=fy;            ans+=E[i].z;        }        }        printf("%.3f\n",ans);    }}