Kruskal算法C++

 

1. #include<iostream>
2. using namespace std;
3. const int n=6; //图的顶点数
4. const int e=10; //图的边数 n-1=<e<=n(n-1)/2 
5. typedef int adjmatrix[n][n]; 
6. typedef struct{ //生成树的边结点
7. int fromvex,endvex; //边的起点与终点
8. int weight; //边上的权
9. }TreeEdgeNode;
10. typedef TreeEdgeNode edgesetT[n-1]; //这是不
11. typedef TreeEdgeNode edgesetG[e]; 
13. void Kruskal(edgesetG GE,edgesetT CT,int n)
14. //利用克鲁斯卡尔算法求边集数组GE所示图的最小生成树,
15. //树中每条边依次存于数组C中
16. {
17. int i,j;
18. adjmatrix s;//用二维数组s作为集合使用，其中每一行元素
19. //s[i][0]~s[i][n-1]用来表示一个集合，若s[i][t]==1
20. //则表示v[t]顶点属于该集合，否则不属于s[i]集合
21. for(i=0;i<n;i++)
22. { //初始化s集合，使每一个顶点分属于不同的集合 
23. for(j=0;j<n;j++)
24. if(i==j)
25. s[i][j]=1;
26. else
27. s[i][j]=0;
28. }//循环执行结束后，使得v[i]顶点属于s[i]集合中
29. int k=1;//k表示待获取的最小生成树中的边数，初值为1
30. int d=0;//d表示GE中待扫描边元素的下标位置，初值为0
31. int m1,m2;//m1和m2用来分别记录一条边的两个顶点所在的集合的序号
32. while(k<n)
33. {//进行n-1次循环，得到最小生成树中的n-1条边
34. /*for(i=0;i<n;i++)
35. {//求出边GE[d]的两个顶点所在集合的序号m1和m2]
36. for(j=0;j<n;j++)
37. {
38. if(GE[d].fromvex==j&&s[i][j]==1)//找起点
39. m1=i;
40. if(GE[d].endvex==j&&s[i][j]==1)//找终点
41. m2=i;
42. } 
44. }*/
45. //可以把上面的重for循环改良成如下的一重for循环
46. for(i=0;i<n;i++)
47. {
48. if(s[i][GE[d].fromvex]==1)//找起点
49. m1=i;
50. if(s[i][GE[d].endvex]==1)//找终点
51. m2=i;
52. }
53. if(m1!=m2)
54. { //若两集合序号不等，则表明GE[d]是生成树中的一条边，
55. //应将它加入到数组C中
56. CT[k-1]=GE[d];
57. //cout<<"CT[<<k-1<<]=<<CT[k-1].
58. k++;
59. for(j=0;j<n;j++)
60. {//合并两个集合，并将另一个置为空集
61. s[m1][j]=s[m1][j]||s[m2][j];
62. s[m2][j]=0;
63. }
64. }
65. d++;//d后移一个位置，以便扫描GE中的下一条边
66. } 
68. }
69. void main()
70. {
71. int i;
72. edgesetG GE;
73. edgesetT CT;
74. cout<<"请按权值小到大输入边集:"<<endl;
75. for(i=0;i<e;i++)//这是不妥的，因为不知道到底有多少边
76. {
77. cin>>GE[i].fromvex>>GE[i].endvex>>GE[i].weight;
78. }
79. Kruskal(GE,CT,n);
80. cout<<"最上生成树的边集为:"<<endl;
81. for(i=0;i<n-1;i++)
82. {
83. cout<<CT[i].fromvex<<" "<<CT[i].endvex<<" "<<CT[i].weight<<endl;
84. } 
86. }
87. /*测试用例一:
88. 请按权值小到大输入边集:
89. 0 4 4
90. 1 2 5
91. 1 3 8
92. 2 3 10
93. 1 5 12
94. 3 5 15
95. 0 1 18
96. 3 4 20
97. 0 5 23
98. 4 5 25
99. 最上生成树的边集为:
100. 0 4 4
101. 1 2 5
102. 1 3 8
103. 1 5 12
104. 0 1 18
105. */