-
题目描述:
-
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
-
输入:
-
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
-
输出:
-
对每个测试用例,在1行里输出最小的公路总长度。
-
样例输入:
-
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
-
样例输出:
-
3 5
这道很简单的最小生成树的例子,我第一直觉用Prim算法做,因为这道题每两点之间都有线相连,所以属于稠密图,用Kruskal算法时间复杂度也是O(n*2)我凭借印象码完代码之后,第一次就提交,结果RunTime,也是醉了。。经典的Prim算法也会超时,检查了一下,发现题目中节点个数也只有100个,最多也只有两层循环,不可能超时啊。因为之前也这样写过,越发觉得九度OJ好烂。。后来东改改,西改改。因为我想知道哪里超时了。后来在我改了一个小时之后发现,还是输入输出的问题,因为这里有大量的点要输入,所以用cout和cin明显就超时了,默默的改用C语言里面的printf和scanf就A了。这件事告诉我们,做OJ的题,涉及到大量的输入,一定要用printf和scanf
#include <stdio.h> #include <string.h> const int MAX = 110; const int M = 100000001; int a[MAX][MAX]; bool visit[MAX]; int dis[MAX]; void Prim(int n){ int i,j; memset(visit,0,sizeof(visit)); for(i=1;i<=n;i++){ dis[i] = a[1][i]; } for(i=1;i<n;i++){ int temp = M; int k = 0; for(j=1;j<=n;j++){ if(!visit[j] && dis[j]<temp){ temp = dis[j]; k = j; } } visit[k] = 1; for(j=1;j<=n;j++){ if(!visit[j] && dis[j] > a[k][j]){ dis[j] = a[k][j]; } } } for(i=2;i<=n;i++) dis[1] += dis[i]; printf("%d\n",dis[1]); } int main(){ int n; while(scanf("%d",&n)){ if(n == 0) break; memset(a,0,sizeof(a)); for(int i=1;i<=n*(n-1)/2;i++){ int x,y,z; scanf("%d%d%d",&x,&y,&z); a[y][x] = a[x][y] = z; } Prim(n); } return 0; }