HDOJ 1875 畅通工程再续

 题目链接：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;
}