kuangbin专题六 POJ2031 Building a Space Station

最新推荐文章于 2022-03-28 10:54:14 发布

原创最新推荐文章于 2022-03-28 10:54:14 发布 · 237 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#poj

最小生成树专栏收录该内容

17 篇文章

订阅专栏

探讨了如何通过计算球体间的相对位置和半径来判断是否需要使用桥梁连接它们，并利用Kruskal算法实现最小生成树求解。

题意：
给你N个球体的x,y,z点和半径，要你去求他们之间是否需要用桥来搭建联系，如果两个球体相接触则不需要，否则需要。
题解：
N个球体之间相连，可以想到最小生成树，那么剩下的就是计算这些球体之间的距离是否大于他们的半径了。
题外话：
POJ的输出%lf好像是因为机器老旧的问题，要弄成%f的，但是别人好像可以的。。。我就蒙了。

//Kruskal
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int MAXN=105*105;
struct node1
{
    double w,x,y,z;
}g[105];
struct node2
{
    int u,v;
    double l;
}map[MAXN];
int fa[105];
int n,k;
bool cmp(node2 c,node2 d)
{
    return c.l<d.l;
}
int find(int p)
{
    return fa[p]==p? p:fa[p]=find(fa[p]);
}
void Kruskal()
{
    double sum=0;
    for(int i=0;i<k;i++)
    {
        int P=find(map[i].u);
        int Q=find(map[i].v);
        if(P!=Q)
        {
            sum+=map[i].l;
            fa[P]=Q;
        }
    }
    printf("%0.3f\n",sum);
}
int main()
{
    while(~scanf("%d",&n),n)
    {
        k=0;
        for(int i=1;i<=n;i++)
        fa[i]=i;
        for(int i=1;i<=n;i++)
        scanf("%lf%lf%lf%lf",&g[i].x,&g[i].y,&g[i].z,&g[i].w);
        for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            if(i==j)
            continue;
            if(sqrt((g[i].x-g[j].x)*(g[i].x-g[j].x)+(g[i].y-g[j].y)*(g[i].y-g[j].y)+(g[i].z-g[j].z)*(g[i].z-g[j].z))<=g[i].w+g[j].w)
            map[k].l=0;
            else
            map[k].l=sqrt((g[i].x-g[j].x)*(g[i].x-g[j].x)+(g[i].y-g[j].y)*(g[i].y-g[j].y)+(g[i].z-g[j].z)*(g[i].z-g[j].z))-g[i].w-g[j].w;
            map[k].u=i;
            map[k].v=j;
        //  printf("%lf %lf\n",sqrt((g[i].x-g[j].x)*(g[i].x-g[j].x)+(g[i].y-g[j].y)*(g[i].y-g[j].y)+(g[i].z-g[j].z)*(g[i].z-g[j].z)),g[i].w+g[j].w);
        //  printf("~~~%lf\n",map[k].l);
            k++;
        }
        sort(map,map+k,cmp);
        Kruskal();
    }
}