POJ2031，Building a Space Station，最小生成树模版题

这道题其实就是最小生成树模版题，关键是要把题意读懂

解释一下题意（看题不用被题面所吓到，题目长不一定难！而且慢慢翻译文章也是一种乐趣！

空间里有很多球型的单位，它们的半径不一定都是一样的，要让它们都连通，（两个单位连通有三种情况：1）它们彼此挨着或者重叠；2）它们之间用一个通道直接连通（通道是连在球的表面）；3）它们通过某个单位间接连通），然后是，它们是空间中的单位，如果A-B通过通道连通，C-D通过通道连通，几何上这两条通道相交，但是不能理解成A-C就连通，换句话说，两个通道不会相交

第一行先输入一个整数N，表示有N个单位，接下来的N行，每行4个小数，前三个表示其圆心x,y,z坐标，最后是半径

输出最小生成树，如果所有点的连通是通过挨着或者重叠的方式，那么输出结果是0.000

思路就是：

写一个函数，用来计算两点之间的距离，然后呢，用距离减去它们半径之和（如果结果是负数，表明它们俩是挨着的或者重叠的，那么距离为0，否则距离就是最后结果）

得到它们任意两个之间的距离后（距离保存在maz二维数组里），就可以用prim算法跑了

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<climits>
using namespace std;
double maz[109][109];
double d[109];
bool vis[109];
double dis(double a[],double b[])
{
double x,y,z,sum;
x = (a[0]-b[0]) * (a[0]-b[0]);
y = (a[1]-b[1]) * (a[1]-b[1]);
z = (a[2]-b[2]) * (a[2]-b[2]);
sum = sqrt((x+y+z)*1.0);
sum = sum - (a[3]+b[3]);
return sum;
}//得到两点之间的距离！
void prim(int n)
{
int i,j,pos;
double sum,t;
for (i = 1; i <= n; ++i)
d[i] = maz[1][i];
memset(vis,false,sizeof(vis));
vis[1] = true;
sum = 0;
for (i = 0; i < n-1; ++i)
{
t = INT_MAX;
for (j = 1; j <= n; ++j)
{
if (!vis[j] && d[j] < t)
{
t = d[j];
pos = j;
}
}
sum += t;
vis[pos] = true;
for (j = 1; j <= n; ++j)
{
if (!vis[j] && d[j] > maz[pos][j])
{
d[j] = maz[pos][j];
}
}
}
printf("%.3lf\n",sum);
}
int main()
{
int t,i,j;
double data[109][9],tem;
while (~scanf("%d",&t) && t)
{
for (i = 1; i <= t; ++i)
{
for (j = 1; j <= t; ++j)
maz[i][j] = i == j ? 0 : INT_MAX;
}//初始化maz数组
for (i = 1; i <= t; ++i)
{
for (j = 0; j < 4; ++j)
scanf("%lf",&data[i][j]);
}//保存原始数据
for (i = 1; i <= t; ++i)
{
for (j = i+1; j <= t; ++j)
{
tem = dis(data[i],data[j]);
if (tem <= 0)
maz[i][j] = maz[j][i] = 0;
else
maz[i][j] = maz[j][i] = tem;
}
}//用两层for循环计算任意两点之间的距离，将最后结果保存在maz数组里
prim(t);//模板prim算法
}
return 0;
}