hrbeu 哈工程 Tunnels

最小生成树裸题，给出n个点的坐标，求MST的权值总和，用prim实现

 

#include <stdio.h>

#include <math.h>

#include <string.h>

#include <algorithm>

#define INF 100000000.0

#define N    110

using namespace std;

double x[N],y[N],dis[N][N],lowcoat[N];

bool vis[N];

int n;



double DIS(int i , int j)

{

    return sqrt( (x[i]-x[j])*(x[i]-x[j]) + (y[i]-y[j])*(y[i]-y[j]) ) ;

}

int main()

{

    int i,j,k,nn;

    double min,SUM;

    while(scanf("%d",&n)!=EOF)

    {

        if(!n) break;

        for(i=1; i<=n; i++)

            scanf("%lf%lf",&x[i],&y[i]);



        for(i=1; i<=n; i++)

            for(j=1; j<=n; j++)

                dis[i][j]=dis[j][i]=DIS(i,j);



        //用prim算法因为没有给出边集数组而是给出了顶点用prim算法更合理

        memset(vis,0,sizeof(vis));

        for(i=1; i<=n; i++)

            lowcoat[i]=dis[1][i];

        lowcoat[1]=0; vis[1]=1;



        SUM=0;

        for(nn=1; nn<n; nn++)  //个数，还有纳入n-1个点

        {

            min=INF; k=1;

            for(i=2; i<=n; i++) 

                if(!vis[i] && lowcoat[i] < min)

                {

                    min=lowcoat[i];

                    k=i;

                }

            vis[k]=1; lowcoat[k]=0; SUM+=min;

            for(i=2; i<=n; i++)  //更新所有的边

                if(!vis[i] && dis[k][i] < lowcoat[i])

                    lowcoat[i]=dis[k][i];

        }



        printf("%.2f\n",SUM);

    }

    return 0;

}