时间:2021-05-20
很久以前就学过最小生成树之Kruskal和Prim算法,这两个算法很容易理解,但实现起来并不那么容易。最近学习了并查集算法,得知并查集可以用于实现上述两个算法后,我自己动手实现了最小生成树算法。
宏观上讲,Kruskal算法就是一个合并的过程,而Prim算法是一个吞并的过程,另外在Prim算法中还用到了一种数据结构——优先级队列,用于动态排序。由于这两个算法很容易理解,在此不再赘述。接下来给出我的源代码。
输入
第一行包含两个整数n和m,n表示图中结点个数,m表示图中边的条数;接下来m行,每一行包含三个整数u,v,w,表示途中存在一条边(u,v),并且其权重为w;为了便于调试,我的程序是从文件中输入数据的;
输出
输出程序选择的最小生成树的权值之和以及每一条边信息;
Kruskal算法:
#include <iostream>#include <vector>#include <algorithm>#include <fstream>using namespace std; struct Edge{ int u; //边连接的一个顶点编号 int v; //边连接另一个顶点编号 int w; //边的权值 friend bool operator<(const Edge& E1, const Edge& E2) { return E1.w < E2.w; }};//创建并查集void MakeSet(vector<int>& uset, int n){ uset.assign(n, 0); for (int i = 0; i < n; i++) uset[i] = i;}//查找当前元素所在集合的代表元int FindSet(vector<int>& uset, int u){ int i = u; while (uset[i] != i) i = uset[i]; return i;}void Kruskal(const vector<Edge>& edges, int n){ vector<int> uset; vector<Edge> SpanTree; int Cost = 0, e1, e2; MakeSet(uset, n); for (int i = 0; i < edges.size(); i++) //按权值从小到大的顺序取边 { e1 = FindSet(uset, edges[i].u); e2 = FindSet(uset, edges[i].v); if (e1 != e2) //若当前边连接的两个结点在不同集合中,选取该边并合并这两个集合 { SpanTree.push_back(edges[i]); Cost += edges[i].w; uset[e1] = e2; //合并当前边连接的两个顶点所在集合 } } cout << "Result:\n"; cout << "Cost: " << Cost << endl; cout << "Edges:\n"; for (int j = 0; j < SpanTree.size(); j++) cout << SpanTree[j].u << " " << SpanTree[j].v << " " << SpanTree[j].w << endl; cout << endl;}int main(){ ifstream in("data.txt"); int n, m; in >> n >> m; vector<Edge> edges; edges.assign(m, Edge()); for (int i = 0; i < m; i++) in >> edges[i].u >> edges[i].v >> edges[i].w; sort(edges.begin(), edges.end()); //排序之后,可以以边权值从小到大的顺序选取边 Kruskal(edges, n); in.close(); system("pause"); return 0;}Prim算法:
#include <iostream>#include <fstream>#include <vector>#include <list>#include <queue>using namespace std;struct Node{ int v; int w; Node(int a, int b) :v(a), w(b){}};struct Edge{ int u; int v; int w; Edge(int a, int b, int c) :u(a), v(b), w(c){} friend bool operator<(const Edge& E1, const Edge& E2) { return E1.w>E2.w; }};int n, m;vector<list<Node>> Adj;priority_queue<Edge> Q; int FindSet(vector<int>& uset, int v){ int i = v; while (i != uset[i]) i = uset[i]; return i;} void Prim(){ int Cost = 0; //用于统计最小生成树的权值之和 vector<Edge> SpanTree; //用于保存最小生成树的边 vector<int> uset(n,0); //用数组来实现并查集 Edge E(0, 0, 0); for (int i = 0; i < n; i++) uset[i] = i; //并查集初始化 for (auto it = Adj[0].begin(); it != Adj[0].end(); it++) Q.push(Edge(0, it->v, it->w)); //根据Prim算法,开始时选取结点0,并将结点0与剩余部分的边保存在优先队列中 //循环中每次选取优先队列中权值最小的边,并更新优先队列 while (!Q.empty()) { E = Q.top(); Q.pop(); if (0 != FindSet(uset, E.v)) { Cost += E.w; SpanTree.push_back(E); uset[E.v] = E.u; for (auto it = Adj[E.v].begin(); it != Adj[E.v].end(); it++) if (0 != FindSet(uset, it->v)) Q.push(Edge(E.v, it->v, it->w)); } } cout << "Result:\n"; cout << "Cost: " << Cost << endl; cout << "Edges:\n"; for (int j = 0; j < SpanTree.size(); j++) cout << SpanTree[j].u << " " << SpanTree[j].v << " " << SpanTree[j].w << endl; cout << endl;}int main(){ ifstream in("data.txt"); int u, v, w; in >> n >> m; Adj.assign(n, list<Node>()); for (int i = 0; i < m; i++) { in >> u >> v >> w; Adj[u].push_back(Node(v,w)); Adj[v].push_back(Node(u,w)); } Prim(); in.close(); system("pause"); return 0;}就实现而言,Kruskal算法比Prim算法更容易,代码更易于理解。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。
本文介绍了最小生成树的定义,Prim算法的实现步骤,通过简单举例实现了C语言编程。1.什么是最小生成树算法?简言之,就是给定一个具有n个顶点的加权的无相连通图,
之前都是看书,大部分也是c++的实现,但是搞前端不能忘了JS啊,所以JS实现一遍这两个经典的最小生成树算法。一、权重图和最小生成树权重图:图的边带权重最小生成树
最小生成树最小生成树(minimumspanningtree)是由n个顶点,n-1条边,将一个连通图连接起来,且使权值最小的结构。最小生成树可以用Prim(普里
在贪婪算法这一章提到了最小生成树的一些算法,首先是Kruskal算法,实现如下:MST.h复制代码代码如下:#ifndefH_MST#defineH_MST#d
最小生成树Prim算法朴素版有几点需要说明一下。1、2个for循环都是从2开始的,因为一般我们默认开始就把第一个节点加入生成树,因此之后不需要再次寻找它。2、l