什么是最小生成树
最小生成树是一副连通加权无向图中一棵权值最小的生成树[维基百科]
常见的应用例子有铺设道路连接所有城市、铺设管道等,目标都是使总长度最短。
求解最小生成树的基本原理
Prim算法和Kruskal算法是求解最小生成树的两种经典算法,这两个算法都是贪心算法。使用到了MST的一个性质:两个集合之间相连的最短的边一定属于两个集合组成的生成树。该性质的详细证明请见算法导论第23章。这里不赘述了。
Prim算法
概述
Prim算法讲点分为两个集合U和V-U,其中V为全部点的集合。每次从V-U中选择一个距离集合V最近的顶点v1加入U,并讲U中对应的顶点标记为v1的父节点。重复该过程,直到U=V
算法步骤
Step1:初始化U为空,并讲minCut[0]初始化为0,minCut其他值初始化为最大值,minCut用于记录V-U中顶点距离V的最小距离。初始化isInMST数组值为false,parent[0]置为-1,表示0节点没有父节点。
Step2:从minCut中选择值最小的节点v1作为下一个加入U的节点,
Step3:更新V-U中和v1相连的节点距离V的最小距离,即更新minCut数组,并将minCut值变小的节点的父节点(parent)标记为v1.
Step4:重复Step2和Step3,直到U=V,即全部节点都已经加入了集合U。
Step5:此时最小生成树的信息记录在了parent数组中
C++实现
vector<int> primMST (vector<vector<int>> &graph)
{
if(graph.size() == 0)
{
return vector<int>();
}
vector<int> parents(graph.size(), -1);
vector<bool> isInMst(graph.size(), false);
vecto