PKU 2031 Building a Space Station

 

题意理解：

         某空间站，有n个球星房间，要保证每两个房间间存在路径使其互通，需要在两个球星房间之间修一下通道，求通道的和的最短长度。

         注意：

                   某个房间可能包含在其它的房间内，这样就不需要在这两个房间间修通道了。题意理解：

 

解题思路：

         prime 最短路径问题。

         用一个数组d[i],表示第i个房间到已知树的最短距离。每次贪心选择min{d[i]},i=1,2,3….;

将第k个房间联通。

         然后更新d[i].

#include<iostream> #include<algorithm> #include<stdio.h> #include<cmath> using namespace std; const int N=105; const int oo=1<<30; typedef struct{double x,y,z,r;}point; point p[N]; double dis[N][N]; double fun(int i,int j) { double d=sqrt( (p[i].x-p[j].x)*(p[i].x-p[j].x)+(p[i].y-p[j].y)*(p[i].y-p[j].y)+(p[i].z-p[j].z)*(p[i].z-p[j].z) ); double r=p[i].r+p[j].r; if(d>r) return d-r; return 0.000000; } double prime(int n) { bool f[N]; double d[N]; double temp,ans=0.000000; int k,s; memset(f,false,sizeof(f)); temp=oo; f[1]=1; for(int i=2;i<=n;i++) d[i]=dis[1][i]; for(int i=1;i<n;i++) { temp=oo; for(int j=2;j<=n;j++) if(temp>d[j]&&f[j]==0) { temp=d[j]; s=j; } k=s; f[k]=1; ans+=temp; for(int j=2;j<=n;j++) if(f[j]==0&&d[j]>dis[k][j]) d[j]=dis[k][j]; } return ans; } int main() { int n; while(scanf("%d",&n)!=EOF&&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); dis[i][i]=oo; } for(int i=1;i<n;i++) for(int j=i+1;j<=n;j++) dis[j][i]=dis[i][j]=fun(i,j); printf("%.3lf/n",prime(n)); } return 0; }