01. 图的最小生成树

1. 最小生成树的定义

在了解「最小生成树」之前，我们需要要先理解「生成树」的概念。

图的生成树（Spanning Tree）：如果无向连通图 G 的一个子图是一棵包含图 G 所有顶点的树，则称该子图为 G 的生成树。生成树是连通图的包含图中的所有顶点的极小连通子图。图的生成树不惟一。从不同的顶点出发进行遍历，可以得到不同的生成树。

换句话说，生成树是原图 G 的一个子图，它包含了原图 G 的所有顶点，并且通过选择图中一部分边连接这些顶点，使得子图中没有环。

生成树有以下特点：

如上图所示，左侧图 $G$ 是包含 $v_{1} \dots v_{6}$ 共 $6$ 个顶点 $7$ 条边在内的有权图。右侧是图 $G$ 的两个生成树，两者都包含了 $6$ 个顶点 $5$ 条边。

最小生成树（Minimum Spanning Tree）：无向连通图 G 的所有生成树中，边的权值之和最小的生成树，被称为最小生成树。

最小生成树除了包含生成树的特点之外，还具有一个特点。

如上图所示，左侧图 $G$ 是包含 $v_{1} \dots v_{6}$ 共 $6$ 个顶点 $7$ 条边在内的有权图。右侧是图 $G$ 的最小生成树，包含了 $6$ 个顶点 $5$ 条边，并且所有边的权值和最小。

为了找到无向图的最小生成树，常用的算法有「Prim 算法」和「Kruskal 算法」。

这两个算法都可以帮助我们找到图中的最小生成树，以满足连接所有顶点的要求同时使得总权重最小。

Prim 算法的算法思想：每次选择最短边来扩展最小生成树，从而保证生成树的总权重最小。算法通过不断扩展小生成树的顶点集合 $M S T$ ，逐步构建出最小生成树。

将图 $G$ 中所有的顶点 $V$ 分为两个顶点集合 $V_{A}$ 和 $V_{B}$ 。其中 $V_{A}$ 为已经加入到最小生成树的顶点集合， $V_{B}$ 是还未加入生成树的顶点集合。
选择起始顶点 $s t a r t$ ，将其加入到最小生成树的顶点集合 $V_{A}$ 中。
从 $V_{A}$ 的顶点集合中选择一个顶点 $u$ ，然后找到连接顶点 $u$ 与 $V_{B}$ 之间的边中权重最小的边。
让上一步中找到的顶点和边加入到 $M S T$ 中，更新 $M S T$ 的顶点集合和边集合。
重复第 $3 \sim 4$ 步，直到 $M S T$ 的顶点集合中包含了图中的所有顶点为止。

Kruskal 算法的算法思想：通过依次选择权重最小的边并判断其两个端点是否连接在同一集合中，从而逐步构建最小生成树。这个过程保证了最终生成的树是无环的，并且总权重最小。

在实际实现中，我们通常使用并查集数据结构来管理顶点的集合信息，以便高效地判断两个顶点是否在同一个集合中，以及合并集合。

将图中所有边按照权重从小到大进行排序。
将每个顶点看做是一个单独集合，即初始时每个顶点自成一个集合。
按照排好序的边顺序，按照权重从小到大，依次遍历每一条边。
对于每条边，检查其连接的两个顶点所属的集合：
1. 如果两个顶点属于同一个集合，则跳过这条边，以免形成环路。
2. 如果两个顶点不属于同一个集合，则将这条边加入到最小生成树中，同时合并这两个顶点所属的集合。
重复第 $3 \sim 4$ 步，直到最小生成树中的变数等于所有节点数减 $1$ 为止。

来源：https://github.com/itcharge/LeetCode-Py