POJ 2031 Building a Space Station 简单计算几何 + 最小生成树

来源：互联网发布：mac版cad布局颜色设置编辑：程序博客网时间：2024/05/22 22:09

如题　　：简单计算几何 + 最小生成树

求出连个cell的距离小于零距离为零否者就是两球心的距离减两球半径

因为边较少所以用邻接矩阵存储最后用一个prim计算最小生成树答案就出来了

注意：POJ在G++中编译的输出浮点型都要用f 不能用lf

下面看代码：

#include<cstdio>#include<cmath>#include<cstring>#define eps 1e-8#define N 105#define MAX 0x3fffffffusing namespace std;struct cell{    double x,y,z,r;}c[N];double Dis(cell a,cell 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  n;double edge[N][N],res,dis[N];bool flag[N];double prim(int s){    for(int i = 0; i <= N; i++)        dis[i] = MAX;    memset(flag,false,sizeof(flag));    dis[s]=0;    flag[s]=true;    for(int i=1; i<=n; i++)    {        if(dis[i]>edge[s][i])            dis[i]=edge[s][i];    }    for(int i=1; i<n; i++)    {        int pos=-1;        double Min=0x3f3f3f3f;        for(int j=1; j<=n; j++)        {            if(!flag[j]&&dis[j]<Min)            {                Min=dis[j];                pos=j;            }        }        if(pos==-1)  return -1;        res += Min;        flag[pos]=true;        for(int j=1; j<=n; j++)        {            if(!flag[j]&&edge[pos][j]<dis[j])                dis[j]=edge[pos][j];        }    }    return res;}int main(){    #ifdef LOCAL    freopen("in.txt","r",stdin);    #endif // LOCAL    int i,j;    while((scanf("%d",&n),n)!=0)    {        for(i = 1;i <=n; i++)            scanf("%lf%lf%lf%lf",&c[i].x,&c[i].y,&c[i].z,&c[i].r);        for(i = 1; i<=n; i++)            for(j = 1; j<=n; j++)        {            if(i == j){edge[i][j] = MAX;continue;}            double d = Dis(c[i],c[j])-c[i].r-c[j].r;            if(d < eps)  edge[i][j] = 0;            else edge[i][j] = d;        }        res = 0;        double result = prim(1);        printf("%.3f\n",result);    }    return 0;}