代码灵感来自WAY_zhong的视频,点击可以查看
原图如下
这个图最终可以生成两个结构不同的最小二叉树,但最终权值都是37。再次感谢这位大佬的视频!
保证没有任何问题,请放心使用
#include <iostream>
#define INF INT_MAX
//建立图的带权邻接矩阵
int Mgraph[9][9] = {
//0, 1, 2, 3, 4, 5, 6, 7, 8
{INF,4, INF,INF,INF,INF,INF,8, INF},//0
{4, INF,8, INF,INF,INF,INF,11, INF},//1
{INF,8, INF,7, INF,4, INF,INF,2 },//2
{INF,INF,7 ,INF,9, 14, INF,INF,INF},//3
{INF,INF,INF,9, INF,10, INF,INF,INF},//4
{INF,INF,4, 14, 10 ,INF,2, INF,INF},//5
{INF,INF,INF,INF,INF,2, INF,1, 6 },//6
{8 ,11 ,INF,INF,INF,INF,1 ,INF,7 },//7
{INF,INF,2, INF,INF,INF,6 ,7 ,INF},//8
};
void prim(int Mgraph[9][9], int v0, int& sum)
{
//准备工作开始
bool selected[9];
int minidest[9], parent[9];
for (int i = 0; i < 9; i++)
{
selected[i] = false;
minidest[i] = INF;
parent[i] = -1;
}
int count = 8, minedge, v=-1;//count是还剩多少个顶点没有被包含,minedge是选出的最小权值边,v是选出的新的顶点
selected[v0] = true;
parent[v0] = v0;
//准备工作结束
while (count)
{
minedge = INF;
for (int i = 0; i < 9; i++)
{
if ((minidest[i] > Mgraph[v0][i]) && (!selected[i]))
{
minidest[i] = Mgraph[v0][i];
parent[i] = v0;
}
if ((minidest[i] < minedge) && (!selected[i]))
{
minedge = minidest[i];
v = i;
}
}
sum += minidest[v];
selected[v] = true;
minidest[v] = INF;
v0 = v;
count--;
}
}
int main()
{
int sum = 0;
prim(Mgraph, 0, sum);
std::cout << sum;
}
下面是kruskal算法的代码,该算法比较简单。这段代码是改自天勤上面的。
#include <iostream>
int getroot(int n, int v[])
{
while (n != v[n])
n = v[n];
return n;
}
typedef struct Road {
int a, b, e;
}Road;
Road road[14] = {
{7,6,1},
{2,8,2},
{6,5,2},
{2,5,4},
{0,1,4},
{6,8,6},
{7,8,7},
{2,3,7},
{1,2,8},
{0,7,8},
{3,4,9},
{4,5,10},
{1,7,11},
{3,5,14},
};
void kruskal(Road road[], int e, int&sum)
{
int a, b; sum = 0;
int v[9];//并查集
for (int i = 0; i < 9; i++)
v[i] = i;
//sort(road, e);//这个函数我就不实现了,就是按权值将road数组从小到大排序
for (int i = 0; i < e; i++)
{
a = getroot(road[i].a, v);
b = getroot(road[i].b, v);
if (a != b)
{
v[b] = a;
sum += road[i].e;
}
}
}
int main()
{
int sum = 0;
kruskal(road, 14, sum);
std::cout << sum;
}