poj2031 Building a Space Station 最小生成树,kruskal

9 篇文章 0 订阅
9 篇文章 0 订阅

题目链接:http://poj.org/problem?id=2031

题目大意:就是给出三维坐标系上的一些球的球心坐标和其半径,搭建通路,使得他们能够相互连通。如果两个球有重叠的部分则算为已连通,无需再搭桥。求搭建通路的最小费用(费用就是边权,就是两个球面之间的距离)。

思路:先处理空间点之间的距离,要注意的是两个球面相交的情况,相交的话距离是0。两球面距离是球心距离减去两个球的半径,相交时距离是算出来是负值,要改为0。其他的就是一个球最小生成树的过程,水题

///2014.7.12
///poj2031

//Accepted    764K    16MS    G++ 1856B   2014-07-12 01:46:33

/*
 *最小生成树,kruskal
 */

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;

int n;
double need;
struct Station
{
    double x,y,z;
    double r;
};
Station station[110];

struct Vdge
{
    int x,y;
    double l;
};
bool cmp(Vdge a,Vdge b){
    return a.l < b.l ;
}
Vdge vdge[5500];
int dSet[110];
int find(int x){
    while( x != dSet[x] )
        x = dSet[x];
    return x;
}
bool judge(int x,int y){
    int fx = find(x);
    int fy = find(y);
    if( fx==fy )
        return false;
    else{
        dSet[fx] = fy;
        return true;
    }
}
double dis(int i,int j){
    double distance;
    double a = pow(station[i].x-station[j].x , 2);
    double b = pow(station[i].y-station[j].y , 2);
    double c = pow(station[i].z-station[j].z , 2);
    distance = sqrt(a+b+c);
    distance -= station[i].r + station[j].r;
    if( distance <= 0 ) distance = 0;
    return distance;
}
void init(){
    for(int i=0 ; i<n ; i++)
        scanf("%lf %lf %lf %lf",&station[i].x,&station[i].y,&station[i].z,&station[i].r);
    for(int i=0 ; i<n ; i++)
        dSet[i] = i;
    need = 0;
    int k = 0;
    for(int i=0 ; i<n ; i++){
        for(int j=0 ; j<n ; j++){
            if( j>i ){
                vdge[k].x = i;
                vdge[k].y = j;
                vdge[k].l = dis(i,j);
                k++;
            }
        }
    }
    sort(vdge,vdge+n*(n-1)/2,cmp);
}
void kruskal(){
    int i=0,j=0;
    while( i<n-1 ){
        if( judge(vdge[j].x,vdge[j].y) )
            need += vdge[j].l , i++;
        j++;
    }
}
int main(){
    // freopen("in","r",stdin);
    // freopen("out","w",stdout);

    while( scanf("%d",&n) && n ){
        init();
        kruskal();
        printf("%.3f\n",need );
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值