POJ-2031 Building a Space Station

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/C

题目大意：三维空间中有N个点，其空间距离为相连的费用，求将其全部相连的最小费用。

解题思路：稠密图的最小生成树，先在每两个点之间连边。再用prim，因为只有100个点Kruskal应该也行。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<cstring>
using namespace std;
struct Node{
	double x,y,z,r;
}p[105];
double dist(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))-a.r-b.r;
} 
double d[102][102];
int n;
double ans;
bool vis[105];
void Prim(){
	ans=0.0;
	memset(vis,0,sizeof(vis));
	int s=1; vis[1]=1;
	for(int h=1;h<n;h++){
		double temp=1e10;int k;
		for(int i=2;i<=n;i++){
			if(!vis[i]&&temp>d[s][i]){
				temp=d[s][i];
				k=i;
			} 
		} 
		vis[k]=1;
		ans+=d[s][k];
		for(int i=2;i<=n;i++)
			if(!vis[i]&&d[s][i]>d[k][i])
				d[s][i]=d[k][i]; 	
	}
} 
int main(){
	while(scanf("%d",&n),n){
		for(int i=1;i<=n;i++){
			scanf("%lf%lf%lf%lf",&p[i].x,&p[i].y,&p[i].z,&p[i].r); 
		} 
		for(int i=1;i<=n;i++)
			for(int j=i+1;j<=n;j++){
				double t=dist(p[i],p[j]);
				if(t<1e-7) d[i][j]=d[j][i]=0;
				else {
					d[i][j]=d[j][i]=t;
				}	
			}
		Prim();
		printf("%.3lf\n",ans);	 
	}
	return 0;
}