最小生成树算法 :Prim算法 和 Kruskal 算法

加权无向图G的生成树的代价是该生成树的所有边的代价之和。下面两个算法都采用了贪心策略。
Kruskal算法描述:设T是构造的最小生成树,初始为空。每次从图中选择一条权值最小的边,如果该边不能与T中已有的边形成环。则将该边加入T。如此循环直到T中含有N-1条边为止。N为节点个数。
伪代码如下:

Prim算法描述:设TV是最小生成树已选顶点集合。T是最小生成树已选边集合。算法首选选择任意一个顶点u加入到TV,然后选择一条代价最小的边(u, v)加入到T。重复以上步骤,使得最后T中含有N-1条边。(注意和Kruskal的区别,Prim算法在任意时刻,T中的都是边形成的都是一棵树。而Kruskal算法则可能在某一时刻形成的是一个森林)
伪代码如下:

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Prim算法: ```java import java.util.ArrayList; import java.util.HashSet; import java.util.List; import java.util.PriorityQueue; import java.util.Set; public class PrimMST { public static List<Edge> primMST(Graph graph) { List<Edge> result = new ArrayList<>(); Set<Integer> visited = new HashSet<>(); PriorityQueue<Edge> pq = new PriorityQueue<>((a, b) -> a.weight - b.weight); visited.add(0); for (Edge e : graph.edges[0]) { pq.offer(e); } while (!pq.isEmpty() && visited.size() < graph.vertices) { Edge e = pq.poll(); if (visited.contains(e.to)) { continue; } visited.add(e.to); result.add(e); for (Edge next : graph.edges[e.to]) { if (!visited.contains(next.to)) { pq.offer(next); } } } return result; } } ``` Kruskal算法: ```java import java.util.ArrayList; import java.util.Collections; import java.util.Comparator; import java.util.List; public class KruskalMST { public static List<Edge> kruskalMST(Graph graph) { List<Edge> result = new ArrayList<>(); UnionFind uf = new UnionFind(graph.vertices); List<Edge> edges = new ArrayList<>(); for (int i = 0; i < graph.vertices; i++) { edges.addAll(graph.edges[i]); } // sort edges by weight Collections.sort(edges, Comparator.comparingInt(a -> a.weight)); for (Edge e : edges) { int root1 = uf.find(e.from); int root2 = uf.find(e.to); if (root1 != root2) { uf.union(root1, root2); result.add(e); } } return result; } } ``` 测试代码: ```java import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.Test; import java.util.List; public class TestMST { @Test public void testPrim() { Graph g = new Graph(5); g.addEdge(0, 1, 2); g.addEdge(0, 3, 6); g.addEdge(1, 3, 8); g.addEdge(1, 2, 3); g.addEdge(1, 4, 5); g.addEdge(2, 4, 7); g.addEdge(3, 4, 9); List<Edge> result = PrimMST.primMST(g); Assertions.assertEquals(result.size(), 4); Assertions.assertTrue(result.contains(new Edge(0, 1, 2))); Assertions.assertTrue(result.contains(new Edge(1, 2, 3))); Assertions.assertTrue(result.contains(new Edge(1, 4, 5))); Assertions.assertTrue(result.contains(new Edge(0, 3, 6))); } @Test public void testKruskal() { Graph g = new Graph(5); g.addEdge(0, 1, 2); g.addEdge(0, 3, 6); g.addEdge(1, 3, 8); g.addEdge(1, 2, 3); g.addEdge(1, 4, 5); g.addEdge(2, 4, 7); g.addEdge(3, 4, 9); List<Edge> result = KruskalMST.kruskalMST(g); Assertions.assertEquals(result.size(), 4); Assertions.assertTrue(result.contains(new Edge(0, 1, 2))); Assertions.assertTrue(result.contains(new Edge(1, 2, 3))); Assertions.assertTrue(result.contains(new Edge(1, 4, 5))); Assertions.assertTrue(result.contains(new Edge(0, 3, 6))); } } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值