图的最小生成树算法:普利姆算法

问题描述

最小生成树的概念:
最小生成树的意思就是,给你很多边和点的图,请你求出使得所有顶点都能联通起来,同时边的权值之和最小的结果,这个结果就是最小生成树。

思路分析

既然要求把所有顶点都连起来,并且边的权值最小,那么,一般就有两种思路,一种是从边出发,每次挑选最小的边,这是克鲁斯卡尔算法,还有一种就是普利姆算法了,从顶点出发,每次挑选出当前可以走的最短路径。
普利姆算法思路详解:
首先,从任意一个节点开始,然后,把当前节点的所有边加入到一个集合:解锁集合。
如何理解解锁集合?解锁集合的意思就是,当前可供选择的所有边,然后,将解锁集合排序,每次从中选取最小的边,然后观察最小的边连接的节点是否已经被处理过,如果被处理过,就继续选择下一个最小的边,如果没被处理过,那么,就相当于发现一个新节点,把新节点加入结果集中,然后,将新节点所有边再加入解锁集合,然后依次重复上述步骤,直到解锁集合遍历完,仍然未发现新节点,说明最小生成树完成了。

代码

    public static Set<Edges> primMST(Graph graph){
    	//优先级队列
        PriorityQueue<Edges> queue=new PriorityQueue<>(new EdgeComparator());
        //保存结果
        HashSet<Edges> result=new HashSet<>();
        //判断某个节点是否已经遍历过
        HashSet<Node> set=new HashSet<>();
        //外层循环解决森林的问题
       for (Node n : graph.nodes.values()){
           if(!set.contains(n)){
               for (Edges edge : n.edges) {
                   queue.add(edge);
               }
               set.add(n);
               while (!queue.isEmpty()){
                   Edges e=queue.poll();
                   Node to=e.to;
                   if(!set.contains(to)){
                        result.add(e);
                        set.add(to);
                        for (Edges edges : to.edges){
                            queue.add(edges);
                        }
                   }
               }
           }
       }
       return result;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值