hdu 1162 prim算法

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

题目大意：图纸上有一些点，要用通过墨水用直线将它们连在一起，求用最少的墨水时所连的长度

思路：典型的最小生成树问题，只是要调用函数自己计算这些点之间的距离

#include<stdio.h> #include<math.h> #define N 105 #define Initmax 1005 double a[N][N],lowcost[N],x[N],y[N]; int s[N],n; double dist(double x1,double y1,double x2,double y2) { return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)); } double prim() { int i,j,u; double sum=0,Max; for(i=1;i<=n;i++) { lowcost[i]=a[1][i]; s[i]=0; } s[1]=1; for(i=1;i<n;i++) { Max=Initmax; for(j=1;j<=n;j++) { if(!s[j] && lowcost[j]<Max) { Max=lowcost[j]; u=j; } } if(Max==Initmax) break; s[u]=1; sum+=lowcost[u]; for(j=1;j<=n;j++) if(!s[j] && lowcost[j]>a[u][j]) lowcost[j]=a[u][j]; } return sum; } int main() { int i,j; double result; while(scanf("%d",&n)!=EOF) { 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++) a[i][j]=dist(x[i],y[i],x[j],y[j]); result=prim(); printf("%.2f/n",result); } return 0; }