poj-2031 Building a Space Statio

最新推荐文章于 2019-09-07 11:01:14 发布

原创最新推荐文章于 2019-09-07 11:01:14 发布 · 622 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#acm #算法

----------生成树专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种计算三维空间中基站间有效距离的方法，并使用Prim算法求解最小生成树来找出连接所有基站的最短路径。通过C++实现，展示了如何处理球面距离并避免负距离情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：点击打开链接

又是一发 G++ WA , C++ AC.

简单题：给出基站的三围坐标及半径。若球面距离小于 0 距离为 0 ，否则为圆心距离减去2个半径

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <string.h>

#define INF 10000000

using namespace std;

double map[110][110],dis[110];
int v[110];

struct node
{
    double x,y,z,r;
}p[110];

double makedis(int i,int j)
{
    double ans = sqrt((p[i].x - p[j].x)*(p[i].x - p[j].x) + (p[i].y - p[j].y)*(p[i].y - p[j].y) + (p[i].z - p[j].z)*(p[i].z - p[j].z));
    ans -= (p[i].r + p[j].r);
    if (ans <= 0.0000000001)
        return 0;
    else
        return ans;
}

double prim(int n)
{
    int k;
    double min,sum=0;
    memset(v,0,sizeof(v));
    memset(dis,0,sizeof(dis));

    for(int i=1;i<=n;i++)
        dis[i]=map[1][i];

    v[1]=1;
    dis[1]=0;

    for(int i=2;i<=n;i++)
    {
        k=-1;
        min = INF;
        for(int j=1;j<=n;j++)
            if(!v[j]&&min>dis[j])
            {
                k=j;
                min=dis[j];
            }

        sum += min;
        v[k]=1;

        for(int j=1;j<=n;j++)
            if(!v[j] && dis[j]>map[k][j])
                dis[j]=map[k][j];
    }
    return sum;
}

int main()
{
    int n,m;
    while(cin>>n)
    {
        if(n == 0)
            break;

        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                if (i != j)
                    map[i][j] = INF;
                else
                    map[i][j] = 0;
            }
        for(int i=1;i<=n;i++)
           {
               cin>>p[i].x>>p[i].y>>p[i].z>>p[i].r;
           }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                if (map[i][j] >= makedis(i,j))
                {
                    map[i][j]  = makedis(i,j);
                    map[j][i]  = makedis(i,j);
                }
            }
        printf("%.3lf\n",prim(n));
    }
    return 0;
}