SDUT 2144 图结构练习——最小生成树

题目链接 ：http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2144

图结构练习——最小生成树

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

输入

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)

剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

 

输出

 每组输出占一行，仅输出最小花费。

示例输入

3 2
1 2 1
1 3 1
1 0

示例输出

2
0

提示

 

来源

 赵利强

示例程序

由于n*n和m的大小差不多，所以这题可以选择Prime算法（n*n<m)，也可以选择Kruskal(n*n>m)

先是Prime算法，代码如下

view plain copy print 如果您复制代码时出现行号，请点击左边的“view plain”后再复制

1. #include <cstdio>  
2. #include <cstring>  
3. #include <algorithm>  
4. #define inf 99999999  
5. using namespace std;
6.   
7. int main()  
8. {  
9.     int n, m;  
10.     while(~scanf("%d %d", &n, &m))  
11.     {  
12.         int a[100+5][100+5];  
13.         for(int i=1;i<=n;i++)  //初始化
14.             for(int j=1;j<=n;j++)  
15.             if(i==j) a[i][j]=0;  
16.             else a[i][j]=inf;  
17.         while(m--)  
18.         {  
19.             int u, v, w;  
20.             scanf("%d %d %d", &u, &v, &w);  
21.             if(a[u][v]>w) a[u][v]=a[v][u]=w;  //注意这里是无向图，并且可能存在重复的u、v信息
22.         }  
23.         int dis[100+5], sum=0;  
24.         bool vis[100+5];  
25.         for(int i=1;i<=n;i++)  
26.             dis[i]=a[1][i];  
27.         memset(vis,0,sizeof(vis));  //设置标记数组vis
28.         vis[1]=1;  
29.         for(int i=1;i<n;i++)  
30.         {  
31.             int min=inf, pos=inf;  
32.             for(int j=1;j<=n;j++)  
33.                 if(!vis[j]&&dis[j]<min)  //查找并记录最小值及其下标，表明此值已经“明确”为最短（意思是没有其他路径可以让它变得更短，也就是无法再进行“松弛”）
34.             {  
35.                 min=dis[j];  
36.                 pos=j;  
37.             }  
38.             if(min>=inf) break;  //如果所有值都已“明确”为最短，结束循环
39.             sum+=min;  
40.             vis[pos]=1;  
41.             for(int j=1;j<=n;j++)  
42.                 if(!vis[j]&&dis[j]>a[pos][j])  //使用该最短路径“松弛”其他路径
43.                    dis[j]=a[pos][j];  
44.         }  
45.         printf("%d\n", sum);  
46.     }  
47.     return 0;  
48. }  

再是Kruskal：

view plain copy print 如果您复制代码时出现行号，请点击左边的“view plain”后再复制

1. #include <cstdio>  
2. #include <cstring>  
3. #include <algorithm>  
4. using namespace std;  
5.   
6. int pre[100+5];  
7.   
8. struct node  
9. {  
10.     int u, v, w;  
11. }boy[10000+5];  
12.   
13. bool cmp(node a, node b)  
14. {  
15.     return a.w<b.w;  
16. }  
17.   
18. int root(int a)  //查找
19. {  
20.     int x=a;  
21.     while(x!=pre[x])  
22.     {  
23.         x=pre[x];  
24.     }
25.     while(a!=x)  //路径压缩
26.     {
27.         int j=pre[a];  
28.         pre[a]=x;  
29.         a=j;  
30.     }  
31.     return x;  
32. }  
33.   
34. int main()  
35. {  
36.     int m, n;  
37.     while(~scanf("%d %d", &n, &m))  
38.     {  
39.         for(int i=1;i<=n;i++)  
40.             pre[i]=i;  
41.         for(int i=0;i<m;i++)  
42.             scanf("%d %d %d", &boy[i].u, &boy[i].v, &boy[i].w);  
43.         sort(boy,boy+m,cmp);  //对结构体按路径花费进行升序排序
44.         int ans=0, cnt=0;  
45.         for(int i=0;i<m;i++)  
46.         {  
47.             if(root(boy[i].u)!=root(boy[i].v))  //合并
48.             {  
49.                 pre[root(boy[i].u)]=root(boy[i].v);  
50.                 ans+=boy[i].w;  
51.                 cnt++;  
52.             }  
53.             if(cnt>=n-1) break;  //只需再联通n-1个节点即可保证全部联通
54.         }  
55.         printf("%d\n", ans);  
56.     }  
57.     return 0;  
58. }