图论- 经典最小天生树算法

用户国营  金牌会员 | 2025-2-17 10:26:13 | 来自手机 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 987|帖子 987|积分 2961

什么是最小天生树

在图中找一棵包罗图中全部节点的树, 且权重和最小的那棵树就叫最小天生树.
如下:右侧天生树的权重和显然比左侧天生树的权重和要小。(但是它并不是最小的,这里只是比较一下不同的树)

Kruskal算法

最小天生树是多少条边的集合(称为mst)
最小天生树中我们必要保证几点:


  • 包罗图中全部节点
  • 形成的布局是树布局(不存在环)
  • 权重和最小
    前面两条很容易利用并查集算法做到
    最后一条这里使用了贪婪算法的头脑:
    将全部边按照权重从小到大排序,从权重最小的边开始遍历,如果这条边和mst中其他边不形成边,则称为mst的一部门,否则不加入.
关键代码实现

想学好这个算法有两个关键的点:

  • 熟悉掌握并查集算法,对于判断是否成环和连通分量如何盘算要熟悉
  • 清楚贪婪头脑和最后如何累积权重并返回.
  1. [/code] [size=4]Prim 最小天生树算法[/size]
  2. 它也是使用贪婪头脑来让天生树的权重尽可能小,它接纳切分的方式去处置处罚.
  3. 简单来说呢就是从一个节点开始,看这个节点有几条路径,然后选一条最短.
  4. 走到下一个节点的时候,同时考虑之前的节点的路径,然后选一条最短的,然后循环这样.
  5. 举例:
  6. [align=center][img]https://i-blog.csdnimg.cn/direct/8ac0d591cd11421a9b09be3381b092d1.png[/img][/align]
  7. 这里固然是在B节点,但是也要考虑A节点的情况.
  8. Prim 算法的逻辑就是这样,每次切分都能找到最小天生树的一条边,然后又可以举行新一轮切分,直到找到最小天生树的全部边为止。
  9. 那代码如何实现呢?
  10. 我们如何盘算横切边有哪些,好比是否可以快速算出 cut({A, B, C}),也就是节点 A, B, C 的全部「横切边」有哪些?
  11. [align=center][img]https://i-blog.csdnimg.cn/direct/d3b827feba7b4f75a8e01999213ac4f2.png[/img][/align]
  12. 是可以的,因为我们发现:
  13.   cut({A, B, C}) = cut({A, B}) + cut({C})
  14. [align=center][img]https://i-blog.csdnimg.cn/direct/6201b91b2c92426da97e5adf377787be.png[/img][/align]
  15. 这个特点使我们用我们写代码实现「切分」和处置处罚「横切边」成为可能:
  16. 在举行切分的过程中,我们只要不断把新节点的邻边加入横切边集合,就可以得到新的切分的全部横切边。
  17. 当然,细心的你肯定发现了,cut({A, B}) 的横切边和 cut({C}) 的横切边中 BC 边重复了。
  18. 不外这很利益理,用一个布尔数组 inMST 辅助,防止重复盘算横切边就行了。
  19. 最后一个题目,我们求横切边的目的是找权重最小的横切边,怎么做到呢?
  20. 很简单,用一个
  21. 优先级队列 存储这些横切边,就可以动态盘算权重最小的横切边了。
  22. 所以接纳inMST和优先级队列可以帮助我们实现这个算法.
  23. 实现如下:
  24. [code]class Prim {
  25.     // 核心数据结构,存储「横切边」的优先级队列
  26.     private PriorityQueue<int[]> pq;
  27.     // 类似 visited 数组的作用,记录哪些节点已经成为最小生成树的一部分
  28.     private boolean[] inMST;
  29.     // 记录最小生成树的权重和
  30.     private int weightSum = 0;
  31.     // graph 是用邻接表表示的一幅图,
  32.     // graph[s] 记录节点 s 所有相邻的边,
  33.     // 三元组 int[]{from, to, weight} 表示一条边
  34.     private List<int[]>[] graph;
  35.     public Prim(List<int[]>[] graph) {
  36.         this.graph = graph;
  37.         this.pq = new PriorityQueue<>((a, b) -> {
  38.             // 按照边的权重从小到大排序
  39.             return a[2] - b[2];
  40.         });
  41.         // 图中有 n 个节点
  42.         int n = graph.length;
  43.         this.inMST = new boolean[n];
  44.         // 随便从一个点开始切分都可以,我们不妨从节点 0 开始
  45.         inMST[0] = true;
  46.         cut(0);
  47.         // 不断进行切分,向最小生成树中添加边
  48.         while (!pq.isEmpty()) {
  49.             int[] edge = pq.poll();
  50.             int to = edge[1];
  51.             int weight = edge[2];
  52.             if (inMST[to]) {
  53.                 // 节点 to 已经在最小生成树中,跳过
  54.                 // 否则这条边会产生环
  55.                 continue;
  56.             }
  57.             // 将边 edge 加入最小生成树
  58.             weightSum += weight;
  59.             inMST[to] = true;
  60.             // 节点 to 加入后,进行新一轮切分,会产生更多横切边
  61.             cut(to);
  62.         }
  63.     }
  64.     // 将 s 的横切边加入优先队列
  65.     private void cut(int s) {
  66.         // 遍历 s 的邻边
  67.         for (int[] edge : graph[s]) {
  68.             int to = edge[1];
  69.             if (inMST[to]) {
  70.                 // 相邻接点 to 已经在最小生成树中,跳过
  71.                 // 否则这条边会产生环
  72.                 continue;
  73.             }
  74.             // 加入横切边队列
  75.             pq.offer(edge);
  76.         }
  77.     }
  78.     // 最小生成树的权重和
  79.     public int weightSum() {
  80.         return weightSum;
  81.     }
  82.     // 判断最小生成树是否包含图中的所有节点
  83.     public boolean allConnected() {
  84.         for (int i = 0; i < inMST.length; i++) {
  85.             if (!inMST[i]) {
  86.                 return false;
  87.             }
  88.         }
  89.         return true;
  90.     }
  91. }
复制代码
Kruskal 和 Prim 算法的区别

特点Kruskal算法Prim算法处置处罚方式从边出发,选择最小的边加入天生树从一个节点出发,逐步扩展天生树实用图的类型适当稀疏图适当稠密图数据布局使用并查集来检测环,处置处罚边集合使用优先队列(最小堆)来选择最小边时间复杂度(O(E \log E))(O(E \log V))边的处置处罚方式对全部边举行排序按节点扩展,逐步选择最小边图的表示方式边列表毗邻矩阵或毗邻表 为什么Prim算法不必要判断成环,但Kruskal必要

Kruskal算法必要查抄是否会成环,因为它是从全局边集合出发逐步加入边,必要判断两个节点是否已经连通。
Prim算法不必要显式查抄环,因为它是从节点逐步扩展天生树的过程,保证了每次连接的边都是新增的,不会成环。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

用户国营

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表