HDOJ 1233
还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others) Total Submission(s): 21620 Accepted Submission(s): 9631
Problem Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。 当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最小的公路总长度。
Sample Input
3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0
Sample Output
3 5
prim算法:
1 #include <stdio.h> 2 #include<string.h> 3 int g[101][101]; 4 int ans; 5 #define inf 0xffffff 6 void prim(int n) 7 { 8 int lowcost[101],used[101],closet[101],i,j,k; 9 memset(used,0,sizeof(used)); 10 for(i=1;i<=n;i++) 11 lowcost[i]=g[i][1], 12 closet[i]=1; 13 used[1]=1; 14 for(i=1;i<n;i++) 15 { 16 j=1; 17 int min=inf; 18 for(k=2;k<=n;k++) 19 if(lowcost[k]<min&&!used[k]) 20 min=lowcost[k],j=k; 21 used[j]=1; 22 ans+=g[j][closet[j]]; 23 for(k=2;k<=n;k++) 24 if(g[k][j]<lowcost[k]&&!used[k]) 25 lowcost[k]=g[k][j], 26 closet[k]=j; 27 } 28 29 } 30 int main() 31 { 32 int n,i,j,a,b,x; 33 while(scanf("%d",&n)!=EOF&&n) 34 { 35 ans=0; 36 for(i=1;i<=n;i++) 37 g[i][i]=0; 38 for(i=1;i<=(n-1)*n/2;i++) 39 { 40 scanf("%d %d %d",&a,&b,&x); 41 g[a][b]=g[b][a]=x; 42 } 43 prim(n); 44 printf("%d\n",ans); 45 } 46 return 0; 47 }