POJ - 2031

题意:给你n个球,如果两个球是接触的那么连接就不花费,如果不是接触的那么就是接触的距离,很容易看出是个最小生成树的。

最近对ACM有点迷茫,不知道为什么,感觉看到每个题都感觉很难,也不愿意去思考,有点脑子疼。。很无奈,不知怎么取舍,毕竟我的专业不是专攻这个的。。哎,有点无奈。

#include <iostream>
#include <math.h>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <iomanip>
using namespace std;
const int maxn = 20000 + 20;
struct node
{
    int come, to;
    double cost;
}edge[maxn];
double x[maxn], y[maxn], z[maxn], r[maxn];
int  fa[maxn];
void init()
{
    for(int i = 0; i < maxn; i++)
        fa[i] = i;
}
int getf(int x)
{
    if(x == fa[x])
        return x;
    return fa[x] = getf(fa[x]);
}
double distance_(int i, int j)
{
    return sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]) + (z[i] - z[j]) * (z[i] - z[j]));
}
int same(int x, int y)
{
    x = getf(x);
    y = getf(y);
    if(x == y)
        return 1;
    fa[x] = y;
    return 0;
}
bool cmp(node p1, node p2)
{
    return p1.cost < p2.cost;
}
double kursual(int sum)
{
    sort(edge, edge + sum, cmp);
    init();
    double ans = 0;
    for(int i = 0; i < sum; i++)
    {
        if(!same(edge[i].come, edge[i].to))
            ans += edge[i].cost;

    }
    return ans;
}
int main()
{
    int n;
    ios::sync_with_stdio(false);
    while(cin >> n && n)
    {
        for(int i = 0; i < n; i++)
            cin >> x[i] >> y[i] >> z[i] >> r[i];
        int sum = 0;
        for(int i = 0; i < n - 1; i++)
        {
            for(int j = i + 1; j < n; j++)
            {

                double dis = distance_(i ,j);
                if(dis <= r[i] + r[j])
                {
                    edge[sum].come = i;
                    edge[sum].to = j;
                    edge[sum++].cost = 0;
                }
                else
                {
                    edge[sum].come = i;
                    edge[sum].to = j;
                    edge[sum++].cost = dis - r[i] - r[j];
                }
            }
        }
        double ans = kursual(sum);
        cout <<setprecision(3) << fixed <<ans << endl;
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值