POJ2031（最小生成树）

来源：互联网发布：ubuntu 16.04输入法编辑：程序博客网时间：2024/05/16 00:52

题目大意：给你n个球的坐标以及半径大小，求MST。如果两球相切或相交，那么间距为0；如果两球相离，那么间距为球心距减去半径。

#include<cstdio>#include<algorithm>#include<cmath>#include<cstring>#include<iostream>#include<string>using namespace std;const int maxn=111111;double x[111],y[111],z[111],r[111];int fa[111];int n,k;struct node{    int u,v;    double w;    bool operator<(const node &a)const    {        return w<a.w;    }}p[maxn];int find_fa(int x){    return (fa[x]==x)?fa[x]:fa[x]=find_fa(fa[x]);}double get_dis(double x1,double y1,double z1,double x2,double y2,double z2){    return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)+(z1-z2)*(z1-z2));}double kruskal(){    double ans=0;    int cnt=0;    sort(p,p+k);    for(int i=0;i<k;i++)    {        int fx=find_fa(p[i].u);        int fy=find_fa(p[i].v);        if(fx!=fy)        {            fa[fx]=fy;            ans+=p[i].w;            cnt++;        }        if(cnt==n-1)            break;    }    return ans;}int main(){    while(scanf("%d",&n)&&n)    {        for(int i=0;i<n;i++)        {            scanf("%lf%lf%lf%lf",&x[i],&y[i],&z[i],&r[i]);            fa[i]=i;        }        k=0;        double dist;        for(int i=0;i<n;i++)        {            for(int j=i+1;j<n;j++)            {                dist=get_dis(x[i],y[i],z[i],x[j],y[j],z[j])-r[i]-r[j];                if(dist<0)                    dist=0;                p[k].u=i;                p[k].v=j;                p[k++].w=dist;            }        }        double res=kruskal();        printf("%.3f\n",res);    }    return 0;}

0 0