C - Building a Space Station poj2031

C - Building a Space Station     poj2031

题意：给定N个圆球通过走廊将其连接起来，找出最短的走廊和。

题解：最小生成树模板，注意走廊最短距离为0。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
#define maxn 1000
struct road
{
	int   a,b;		
	double cost;
}r[100000+5]; //初始化 
struct cell
{
	double x,y,z,r;
}c[100+5]; 		//初始化 
bool com(road a,road b)
{
	return a.cost<b.cost;
}
int  parent[100+5];//初始化 
 
int find(int x )
{
	int r=x;         
    while(r!=parent[r])   //寻找根结点
        r=parent[r];
    int i=x,j;
    while(i!=r){       //压缩路径
        j=parent[i];
        parent[i]=r;
        i=j;
    }
    return r;
}

/*int  find(int x)
{
	while(x!=parent[x])
	{
		x=parent[x];
	}
	return x;
}*/
bool Union(int a,int b)
{
	int  x=find(a);
	int  y=find(b);
	if(x!=y)
	{
		parent[x]=y;
		return true;
	}
	return false;
}
int main()
{
	int n;
	 while(scanf("%d",&n)!=EOF)
	 {
        if(n==0) break;
		int k=1;		///边的数量
		for(int i=1;i<=n;i++)
		{
			scanf("%lf%lf%lf%lf",&c[i].x,&c[i].y,&c[i].z,&c[i].r);
			for(int j=1;j<i;j++)
			{
				double d = sqrt((c[i].x-c[j].x)*(c[i].x-c[j].x)+(c[i].y-c[j].y)*(c[i].y-c[j].y)+(c[i].z-c[j].z)*(c[i].z-c[j].z))-c[i].r-c[j].r;
				r[k].cost=d;
				r[k].a=i;
				r[k].b=j;
				k++;
			}
		}
		sort(r+1,r+1+k-1,com);
		for(int i=1;i<=n+1;i++)
		{
			parent[i]=i;
		}
		int con=0;
		double ans=0;
		for(int i=1;i<k;i++)
		{
			if(Union(r[i].a,r[i].b))
			{
				if (r[i].cost>0)
					ans+=r[i].cost;
				con++;
				if(con==n-1)
				break;
			}
		}
		printf("%.3lf\n",ans);
	}
	return 0;
}