hdu 1301 prim算法

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

题目大意：多年前在村庄间修建了一些路，好多是冗余的，加大了路的负担，现在要减少一些路，只剩余下一些主大道，使村庄间直接或间接可达就行，求路最小的负担。

思路：要使村庄间可达，n个村庄只需n-1条路就行，故为最小生成树的问题，用prim算法解决，要注意的是对输入格式的处理，将大写字母转换成数字表示村庄就可以了

#include<stdio.h> #include<math.h> #define N 28 #define Initmax 1005 int a[N][N],lowcost[N]; int s[N],n; int prim() { int i,j,u,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,result,k,cost; char ch; while(scanf("%d",&n),n!=0) { getchar(); //刚写时，忘记了初始化a[][]，致使测试实例通不过，还调试了好长时间 for(i=1;i<=n;i++) for(j=1;j<=n;j++) a[i][j]=Initmax; for(i=1;i<n;i++) { scanf("%c %d",&ch,&k); getchar(); while(k--) { scanf("%c %d",&ch,&cost); getchar(); j=ch-'A'+1; a[i][j]=a[j][i]=cost; } } result=prim(); printf("%d/n",result); } return 0; }