HDU4463（prim）

题意：给定n个坐标，求最小生成树。

先将nike和apple之间的边加进来，并更新mat[nike][apple]=0,再贪心即可。

或者：先把vis[ nike ]=vis[ apple ]=1,但是同时要更新dis[ i ]=min( mat[apple][i],mat[nike][i]);

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn = 255;
const int inf = 999999;
int vis[ maxn ];
double mat[ maxn ][ maxn ],dis[ maxn ];
struct node{
    int x,y;
}a[ maxn ];
int n;
int nike,apple;

double dist( int x1,int y1,int x2,int y2 ){
    double ans=(x1-x2)*(x1-x2)*1.0+(y1-y2)*(y1-y2)*1.0;
    return sqrt( ans );
}

double prim( ){
    double sum=0;
    sum+=mat[ nike ][ apple ];
    vis[ nike ]=vis[ apple ]=1;
    for( int i=1;i<=n;i++ )
        dis[ i ]=min(mat[ apple ][ i ],mat[ nike ][ i ]);
    for( int i=0;i<n;i++ ){
        int k=-1;
        double max;
        max=inf*1.0;
        for( int j=1;j<=n;j++ ){
            if( vis[ j ]==0 && max>dis[ j ] ){
                k=j,max=dis[ j ];
            }
        }
        if( k==-1 ) return sum;
        vis[ k ]=1;
        sum+=max;
        for( int j=1;j<=n;j++ ){
            if( vis[ j ]==0 && dis[ j ]>mat[ k ][ j ] ){
                dis[ j ]=mat[ k ][ j ];
            }
        }
    }
    return sum;
}

int main(){
    while( scanf("%d",&n)!=EOF,n ){
        scanf("%d%d",&nike,&apple);
        memset( vis,0,sizeof( vis ));
        for( int i=1;i<=n;i++ )
            scanf("%d%d",&a[ i ].x,&a[ i ].y),a[i].x+=100,a[i].y+=100;
        for( int i=1;i<=n;i++ ){
            for( int j=1;j<=n;j++ ){
                if( i!=j )
                    mat[ i ][ j ] = dist( a[ i ].x,a[ i ].y,a[ j ].x,a[ j ].y );
                else 
                    mat[ i ][ j ]=inf*1.0;
            }
        }
        double ans=prim();
        printf("%.2lf\n",ans);
    }
    return 0;
}