HDU--1233最小生成树之kruskal算法

来源：互联网发布：汉字笔顺查询软件编辑：程序博客网时间：2024/06/11 18:22

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 39839 Accepted Submission(s): 18090

Problem Description

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

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

31 2 11 3 22 3 441 2 11 3 41 4 12 3 32 4 23 4 50

Sample Output

35
不知道什么原因，截图的方式不行了，，知道拷贝过来了，开始被TLE三次，我以为是findfa函数递归要了时间，后来仔细一瞧，原来是pre[i]，老大函数从1到n*(n-1)/2了，原本只要到100就行了，白白浪费了时间
#include <iostream>#include <stdio.h>#include <string.h>#define INF 0x3f3f3f3f#include <algorithm>using namespace std;int pre[105];struct sb{int a,b,c;}map[5000];int cmp(sb x,sb y){return x.c<y.c;}int findfa(int a){if(pre[a]!=a) pre[a]=findfa(pre[a]);return pre[a];}void link(int a,int b){int fa=findfa(a);int fb=findfa(b);if(fa!=fb) pre[fa]=fb;}int kuraskal(int n){sort(map+1,map+n+1,cmp);int i;int sum=0;for(i=1;i<=n;i++){if(findfa(map[i].a)!=findfa(map[i].b)){sum+=map[i].c;link(map[i].a,map[i].b);}}return sum;}int main(){int t;while(scanf("%d",&t)!=EOF&&t){int i,n;n=t*(t-1)/2;//memset(map,INF,sizeof(map));for(i=1;i<=n;i++){scanf("%d %d %d",&map[i].a,&map[i].b,&map[i].c);//if(map[a][b]>c) map[a][b]=map[b][a]=c;}for(i=1;i<=t;i++) pre[i]=i;int sum=kuraskal(n);printf("%d\n",sum);}return 0;}

0 0