畅通工程再续算法思想及代码

源程序：http://acm.hdu.edu.cn/showproblem.php?pid=1875

算法思想：很明显这是一道关于最小生成树的题目，又因为边的数目远远大于点的数目，故采用prim算法。
（1）首先指定一个顶点v将其加入U中（设置len[i]=0）.
（2）初始化每点到U的距离：算出其他点到点v的距离，若距离大于1000或小于10，则该路肯定不能修，可设置为这两点间没有路（设置的该边的距离为1000）。
（3）找出最小的边以及对应的不在U的点k，将点k加入U。
（4）更新其它点到U的距离：若某个点到点k的距离小于到前面算出的到U的距离，则将到U的距离等同为到k的距离
（5）重复（3），（4）一直到找到n-1条边，若中途发现没有符合条件的边（既所有边到U距离为1000），则提前结束程序，输出oh！。
#include
#include
#define maxdouble1000
int n,m;
doublelen[101],sum;//len[i]为点到U的距离，sum为所需路费
struct point{
int x;
int y;
};
struct pointp[101];

double dis(point p1,pointp2){//两点间距离
returnsqrt(0.0+(p2.x-p1.x)*(p2.x-p1.x)+(p2.y-p1.y)*(p2.y-p1.y));
}

int Min(doublelen[101]){算出最小边且不在U的点
double s=maxdouble;
int k=-1,i;
for(i=0;i
if(len[i]0.0){
s=len[i];
k=i;
}
}
if(k!=-1)
sum+=s*100;//如果存在一条符合条件的边，将其加进来。
return k;
}

void prim(int k){
int i,j;
for(i=0;i
if(i!=k){//初始化每点到U的距离
if(dis(p[k],p[i])<=1000&&dis(p[k],p[i])>=10)
len[i]=dis(p[k],p[i]);
else
len[i]=maxdouble;
}
}
len[k]=0;//将改点加入U
for(i=1;i
int t=Min(len);
if(k==-1){//如果此次找不到符合条件的边，提前结束。
sum=0.0;
break;
}
len[t]=0;将改点加入U

for(j=0;j
if(dis(p[t],p[j])=10)
len[j]=dis(p[t],p[j]);
}
}
}

int main(){
int i;
scanf("%d",&m);
while(m--){
sum=0;//初始化sum
scanf("%d",&n);
for(i=0;i
scanf("%d%d",&p[i].x,&p[i].y);
prim(0);//指定0为初始点
if(sum!=0.0)//无法形成最小生成树
printf("%.1lf\n",sum);
else
printf("oh!\n");
}
return 0;
}