POJ2031 prim

题意：给定一些cell 和 他们的xyz坐标 半径r

求把所有的cell连接起来的最小花费

prim。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<queue>
using namespace std;
const int maxn = 105;
const int maxm = 20005;
const double inf = 99999999;
double mat[ maxn ][ maxn ];
struct node2{
    double x,y,z,r;
}a[ maxn ];

double dist( node2 a,node2 b ){
    double sum;
    sum=(a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)+(a.z-b.z)*(a.z-b.z);
    return sqrt( sum );
}

int vis[ maxn ];
double dis[ maxn ];

double prim( int n ){
    for( int i=0;i<n;i++ ){
        vis[ i ]=0;
        dis[ i ]=mat[ 0 ][ i ];
    }
    vis[ 0 ]=1;
    double sum=dis[ 0 ];
    for( int i=0;i<n;i++ ){
        int k;
        double t=inf;
        for( int j=0;j<n;j++ ){
            if( vis[ j ]==0&&t>dis[ j ] ){
                k=j;
                t=dis[ j ];
            }
        }
        if( t==inf )
            break;
        vis[ k ]=1;
        sum+=t;
        for( int j=0;j<n;j++ ){
            if( vis[ j ]==0&&dis[ j ]>mat[ k ][ j ] ){
                dis[ j ]=mat[ k ][ j ];
            }
        }
    }
    return sum;
}

int main(){
    int n;
    while( scanf("%d",&n),n ){
        for( int i=0;i<n;i++ )
            scanf("%lf%lf%lf%lf",&a[ i ].x,&a[ i ].y,&a[ i ].z,&a[ i ].r);
        for( int i=0;i<n;i++ ){
            for( int j=i+1;j<n;j++ ){
                double tmp=dist( a[i],a[j] );
                if( tmp>(a[i].r+a[j].r) ){
                    mat[i][j]=mat[j][i]=tmp-a[i].r-a[j].r;
                }
                else{
                    mat[i][j]=mat[j][i]=0;
                }//判断是否加边，加多长的边
                //printf("mat[%d][%d]:%lf\n",i,j,mat[i][j]);
            }
        }
        double ans=prim( n );
        printf("%.3lf\n",ans);
    }
    return 0;
}