[原创]HDU 1233 又是畅通工程 [kruskal]+[prim]【最小生成树入门】
2016-04-17 19:51:34 Tabris_ 阅读数：298
博客爬取于2020-06-14 22:44:33
以下为正文
版权声明：本文为Tabris原创文章，未经博主允许不得私自转载。
https://blog.csdn.net/qq_33184171/article/details/51174895
题目链接 :http://acm.hdu.edu.cn/showproblem.php?pid=1233
还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 37241 Accepted Submission(s): 16762
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
Hint
Hint
Huge input, scanf is recommended.
最简单的最小生成树 入门
就不赘述了
直接上代码
kruskal
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
| # include <iostream>
# include <string.h>
# include <stdio.h>
# include <math.h>
# include <algorithm>
using namespace std;
int pre[1000];
int n,m,mintree;
struct node
{
int a,b,l;
} c[10200];
void init(int n)
{
for(int i=1; i<=n; i++)
pre[i]=i;
return ;
}
int findi(int x)
{
int r=x;
while(r!=pre[r])
r=pre[r];
int i=x,j;
while(i!=j)
{
j=pre[i];
pre[i]=r;
i=j;
}
return r;
}
int cmp(node a,node b)
{
return a.l<b.l;
}
void kruskal()
{
int fx,fy;
for(int i=0; i<m; i++)
{
fx=findi(c[i].a),fy=findi(c[i].b);
if(fx!=fy)
{
mintree+=c[i].l;
pre[fy]=fx;
}
}
return ;
}
int main()
{
while(~scanf("%d",&n))
{
if(n==0) break;
init(n);
m=n*(n-1)/2;
mintree=0;
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].l);
}
sort(c,c+m,cmp);
kruskal();
printf("%d\n",mintree);
}
return 0;
}
|
prim
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
| # include <iostream>
# include<stdio.h>
# include<string.h>
# include<math.h>
const int N = 110;
const int INF = 0x7ffffff;
using namespace std;
int v[N],d[N],mp[N][N],n;
int prim()
{
for(int i=1;i<=n;i++) d[i]=INF,v[i]=0;
d[1]=0;
for(int i=0;i<n;i++)
{
int mmin=INF,t;
for(int j=1;j<=n;j++)
if(!v[j]&&mmin>d[j])
mmin=d[t=j];
v[t]=1;
for(int j=1;j<=n;j++)
if(!v[j])
d[j]=min(d[j],mp[t][j]);
}
int ans=0;
for(int i=1;i<=n;i++) ans+=d[i];
return ans;
}
int main()
{
while(~scanf("%d",&n)&&n)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
mp[i][j]=INF;
int m=n*(n-1)/2;
for(int i=0;i<m;i++)
{
int u,v,t;
scanf("%d%d%d",&u,&v,&t);
if(mp[u][v]>t) mp[u][v]=mp[v][u]=t;
}
cout<<prim()<<endl;
}
}
|
wechat
