最小生成树 — Prim算法

同Kruskal算法一样,Prim算法也是最小生成树的算法,但与Kruskal算法有较大的差别。
Prim算法整体是通过"解锁" + "选中"的方式,点 -> 边 -> 点 -> 边
因为是最小生成树,所以针对的也是无向图 ,所以可以随意选取一个点作为进入点,通过解锁这个点,可以获得从这个点出去的所有边,在通过这些边中权重最小的边解锁其他的点。如此反复。直到最小生成树的形成。

如图所示:

左侧为原始图,从a点出发(哪个点都可以,假设从a),解锁了a点(解锁的点画圈),并且解锁了从a点直接出发权重为1,2,9的三条边(边解锁为虚线),根据权重选择1的边(选择具体边改颜色)。并解锁了b点。

通过解锁的b点,可解锁权重1,3,4,9的边,此时bd边的权重最小为1,所以解锁了d的点。

解锁d后,d直接出来的边4也会进行解锁。再次选择权重较小的为2,但是此时d已经解锁过了,所以不考虑2,再次选择be为3的边解锁。

此时解锁后图形如上面所示,e点解锁后会解锁权重6、7的边。

此时所有的边都已经解锁,选择权重小的边,并且不会形成环的点,进行解锁。

最终去掉所有没被选择的边,剩余的就是最小生成树。

代码实现

基于上面图解是代码实现。点 > 边 -> 点 -> 边的解锁方式。

最外层的for循环可防"森林"。 a -> b c ->d e->f,a可以找到b,c可以找到d, e可以找到f。但是a c e之间互相没关系。

java 复制代码
public static class EdgeComparator implements Comparator<Edge> {

        @Override
        public int compare(Edge o1, Edge o2) {
            return o1.weight - o2.weight;
        }
    }

    public static Set<Edge> primMST(Graph graph) {
        //放入PriorityQueue中,并根据边的权重进行排序
        PriorityQueue<Edge> priorityQueue = new PriorityQueue<>(new EdgeComparator());
        //解锁的点
        Set<Node> setNodes = new HashSet<>();
        //构成最小生成树的所有边
        Set<Edge> result = new HashSet<>();

        //遍历图集中所有的点
        for (Node node : graph.nodes.values()) {
            //如果没解锁
            if (!setNodes.contains(node)) {
                setNodes.add(node);
                //将点的所有的边,放到PriorityQueue中排序
                for (Edge edge : node.edges) {
                    priorityQueue.add(edge);
                }

                while (!priorityQueue.isEmpty()) {
                    Edge edge = priorityQueue.poll();
                    //获取到这个边连接的to点
                    Node toNode = edge.to;
                    if (!setNodes.contains(edge.to)) {
                        //解锁to点
                        setNodes.add(toNode);
                        result.add(edge);
                        //并且将to点所有的边也都放到Queue中
                        for (Edge nextEdge : toNode.edges) {
                            priorityQueue.add(nextEdge);
                        }
                    }
                }
            }
            //如果防森林,就不break 
            break;
        }
        return result;
    }
相关推荐
.生产的驴18 分钟前
SpringBoot 封装统一API返回格式对象 标准化开发 请求封装 统一格式处理
java·数据库·spring boot·后端·spring·eclipse·maven
猿周LV26 分钟前
JMeter 安装及使用 [软件测试工具]
java·测试工具·jmeter·单元测试·压力测试
知来者逆28 分钟前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
晨集28 分钟前
Uni-App 多端电子合同开源项目介绍
java·spring boot·uni-app·电子合同
时间之城30 分钟前
笔记:记一次使用EasyExcel重写convertToExcelData方法无法读取@ExcelDictFormat注解的问题(已解决)
java·spring boot·笔记·spring·excel
阿让啊32 分钟前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北33 分钟前
力扣-160.相交链表
算法·leetcode·链表
椰羊~王小美37 分钟前
LeetCode -- Flora -- edit 2025-04-25
java·开发语言
凯酱1 小时前
MyBatis-Plus分页插件的使用
java·tomcat·mybatis
程序员总部1 小时前
如何在IDEA中高效使用Test注解进行单元测试?
java·单元测试·intellij-idea