一种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
    评论
以下是C++实现Prim算法Kruskal算法求最小生成树的示例代码: Prim算法: ```c++ #include <iostream> #include <vector> #include <queue> #include <cstring> using namespace std; const int MAXN = 1005; const int INF = 0x3f3f3f3f; int n; int g[MAXN][MAXN]; int d[MAXN]; bool vis[MAXN]; void prim(int s) { memset(d, INF, sizeof(d)); memset(vis, false, sizeof(vis)); d[s] = 0; priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq; pq.push(make_pair(d[s], s)); while (!pq.empty()) { int u = pq.top().second; pq.pop(); if (vis[u]) continue; vis[u] = true; for (int v = 1; v <= n; v++) { if (g[u][v] < INF && !vis[v] && g[u][v] < d[v]) { d[v] = g[u][v]; pq.push(make_pair(d[v], v)); } } } } int main() { int m; cin >> n >> m; memset(g, INF, sizeof(g)); for (int i = 1; i <= m; i++) { int u, v, w; cin >> u >> v >> w; g[u][v] = g[v][u] = w; } prim(1); int ans = 0; for (int i = 1; i <= n; i++) { if (d[i] == INF) { cout << "No solution!" << endl; return 0; } ans += d[i]; } cout << ans << endl; return 0; } ``` Kruskal算法: ```c++ #include <iostream> #include <algorithm> #include <vector> using namespace std; const int MAXN = 1005; int n; int fa[MAXN]; struct Edge { int u, v, w; bool operator<(const Edge& e) const { return w < e.w; } } edges[MAXN * MAXN]; int find(int x) { if (x == fa[x]) return x; return fa[x] = find(fa[x]); } int kruskal() { int ans = 0, cnt = 0; sort(edges, edges + n * (n - 1) / 2); for (int i = 1; i <= n; i++) fa[i] = i; for (int i = 0; i < n * (n - 1) / 2; i++) { int u = edges[i].u, v = edges[i].v, w = edges[i].w; int fu = find(u), fv = find(v); if (fu != fv) { fa[fu] = fv; ans += w; cnt++; if (cnt == n - 1) break; } } return ans; } int main() { int m; cin >> n >> m; for (int i = 0; i < m; i++) { int u, v, w; cin >> u >> v >> w; edges[i] = { u, v, w }; } cout << kruskal() << endl; return 0; } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值