题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1875
//HDU 1875 AC #include <iostream> #include <cmath> using namespace std; #define MAX 100010 #define LEN 105 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; } } if(min==MAX){ return -1; } 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,nCase; cin>>nCase; double result;//总价 while(nCase--){ 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]); if(len>=10 && len<=1000){//长度有限制 map[i][j]=map[j][i]=100*len;//要*100 } } } result=prim(n); if(result==-1){ cout<<"oh!"<<endl; } else{ printf("%.1lf\n",result); } } return 0; }