一种prim算法和kruskal算法的c++实现,可编译通过

代码灵感来自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;
} 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值