poj 2031 Kruskal

39 篇文章 0 订阅
3 篇文章 0 订阅
//计算几何+最小生成树
//注意精度
//若两单元园心距离小与半径之和,则距离为0
//反之,为距离-半径之和

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

const int MAXN = 101;
const int MAXM = 500000;
int n;
int M;
double cost;
double eps = 0.0000001;

struct edge
{
	int u;
	int v;
	double w;
}edges[MAXM];

struct unit
{
	double x;
	double y;
	double z;
	double r;
}units[MAXN];

int cmp(const void *a,const void *b)
{
	edge aa = *(const edge *)a;
	edge bb = *(const edge *)b;
	return aa.w - bb.w;
}

int Father[MAXN];
int Rank[MAXN];

void Make_Set()
{
	for(int i = 0; i < MAXN; i++)
	{
		Father[i] = i;
		Rank[i] = 1;
	}
}

int Find(int x)
{
	while(x != Father[x])
		x = Father[x];
	return x;
}

void Union(int x,int y)
{
	x = Find(x);
	y = Find(y);
	if(x == y)
		return;
	if(Rank[x] > Rank[y])
	{
		Father[y] = x;
		Rank[x] += Rank[y];
	}
	else
	{
		Father[x] = y;
		Rank[y] += Rank[x];
	}
}



double distance(double x1,double y1,double z1,double x2,double y2,double z2)
{
	return sqrt((x2-x1)*(x2-x1)+(y2-y1)*(y2-y1)+(z2-z1)*(z2-z1));
}


void Kruskal()
{
	int u,v;
	Make_Set();
	for(int i = 0; i < M; i++)
	{
		u = Find(edges[i].u);
		v = Find(edges[i].v);
		if(u != v)
		{
			cost += edges[i].w;
			Union(u,v);
		}
	}
}

int main()
{
	while(cin>>n)
	{
		if(n == 0)
			break;
		for(int i = 0; i < n; i++)
		{
			cin>>units[i].x>>units[i].y>>units[i].z>>units[i].r;
		}
		M = 0;
		cost = 0;
		for(int i = 0; i < n-1; i++)
		{
			for(int j = 1; j < n; j++ )
			{
				double dis = distance(units[i].x,units[i].y,units[i].z,units[j].x,units[j].y,units[j].z);
				edges[M].u = i;
				edges[M].v = j;
				if((dis - (units[i].r + units[j].r)) < eps)
				{
					edges[M].w = 0;
				}
				else
				{
					edges[M].w = dis - (units[i].r + units[j].r);    
					//cout<<dis<<endl;

				}
				M++;
			}
		}
		qsort(edges,M,sizeof(edges[0]),cmp);
		Kruskal();
		printf("%.3f\n",cost);
	}
	return 0;
}

 

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值