您的位置:首页 > 其它

九度OJ 1017

2016-06-01 16:01 232 查看
题目描述:
    某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
输入:

    测试输入包含若干测试用例。每个测试用例的第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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: