链接:https://vjudge.net/problem/POJ-2031#author=0
题意:
三维空间,给n个圆心x,y,z,半径r的圆,求最短的连线。
接触不需要连。
思路:
求距离,接触权值为0,不接触为权值长度减半径。
代码:
#include <iostream>
#include <memory.h>
#include <string>
#include <istream>
#include <sstream>
#include <vector>
#include <stack>
#include <algorithm>
#include <map>
#include <queue>
#include <math.h>
using namespace std;
typedef long long LL;
const int MAXM = 10000+10;
const int MAXN = 100+10;
struct Node
{
double _x,_y,_z,_r;
}node[MAXN];
struct Path
{
int _l,_r;
double _value;
bool operator < (const Path & that)const{
return this->_value < that._value;
}
}path[MAXM];
double Get_Len(Node a,Node b)
{
return sqrt((a._x-b._x)*(a._x-b._x) + (a._y-b._y)*(a._y-b._y) + (a._z-b._z)*(a._z-b._z));
}
int Father[MAXN];
int Get_F(int x)
{
return Father[x] = (Father[x] == x) ? x : Get_F(Father[x]);
}
int main()
{
int n;
while (cin >> n && n)
{
for (int i = 1;i<=n;i++)
Father[i] = i;
for (int i = 1;i<=n;i++)
cin >> node[i]._x >> node[i]._y >> node[i]._z >> node[i]._r;
int pos = 0;
for (int i = 1;i<=n;i++)
{
for (int j = 1;j<=n;j++)
{
double len = Get_Len(node[i],node[j]);
double r = node[i]._r + node[j]._r;
path[++pos]._l = i;
path[pos]._r = j;
if (len <= r)
path[pos]._value = 0;
else
path[pos]._value = len - r;
}
}
sort(path+1,path+1+pos);
double sum = 0;
for (int i = 1;i <= pos;i++)
{
int tl = Get_F(path[i]._l);
int tr = Get_F(path[i]._r);
if (tl != tr)
{
Father[tl] = tr;
sum += path[i]._value;
}
}
printf("%.3lf
",sum);
}
return 0;
}