HDOJ 1162 Eddy's picture

 

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1162

 

源代码：

//HDOJ 1162 AC

#include <iostream>

#include <cmath>

 

using namespace std;

#define MAX 99999

#define LEN 101

double dist[LEN];

double map[LEN][LEN];

bool isvisited[LEN];

 

//点的结构体

typedef struct Point{

       double x;

       double y;

}Point;

 

//初始化

void init(){

       int i,j;

       for(i=0;i<LEN;i++){

              for(j=0;j<LEN;j++){

                     if(i==j) map[i][j]=0;   //对a[][]进行初始化，一般都要；

                     map[i][j]=MAX;

              }

       }

}

 

//计算两点距离

double caculteD(Point a,Point b){

       double len;

       len=sqrt(1.0*(a.x-b.x)*(a.x-b.x)+1.0*(a.y-b.y)*(a.y-b.y));

       return len;

}

 

//prim算法

double prim(int n){

       int i,j,min,pos;

       double sum=0;

       memset(isvisited,false,sizeof(isvisited));

 

       //初始化

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

              dist[i]=map[1][i];

       }

 

       //从1开始

       isvisited[1]=true;

       dist[1]=MAX;

 

       //找到权值最小点并记录下位置

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

              min=MAX;

              //pos=-1;

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

                     if(!isvisited[j] && dist[j]<min){

                            min=dist[j];

                            pos=j;

                     }

              }      

              sum+=dist[pos];//加上权值

              isvisited[pos]=true;

 

              //更新权值

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

                     if(!isvisited[j] && dist[j]>map[pos][j]){

                            dist[j]=map[pos][j];

                     }

              }

       }      

       return sum;

}

 

int main(){

       Point p[105];

    int i,j,n;

       while(cin>>n){

              init();     

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

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

              }

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

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

                            double len;

                len=caculteD(p[i],p[j]);

                map[i][j]=map[j][i]=len;

            }

              }

              printf("%.2lf\n",prim(n));

       }

        return 0;

}