牛客网AB14 最小生成树

最新推荐文章于 2024-09-14 22:57:21 发布

what is an object

最新推荐文章于 2024-09-14 22:57:21 发布

阅读量137

点赞数

分类专栏：算法文章标签：蓝桥杯 java 算法

本文链接：https://blog.csdn.net/newOneObject/article/details/125212980

版权

算法专栏收录该内容

278 篇文章 1 订阅

订阅专栏

该篇博客介绍了一种实现Prim算法的方法，用于寻找给定图中的最小生成树。作者给出了详细的Java代码实现，包括初始化数据结构、排序边集、逐步构建最小生成树的过程，直至所有点都被包含在内。此算法适用于解决图论问题，优化网络连接成本。

摘要由CSDN通过智能技术生成



import java.util.*;

/**
 * @author xnl
 * @Description:
 * @date: 2022/6/9   21:43
 */
public class Solution {
    public static void main(String[] args) {
        Solution solution = new Solution();
        int[][] cost = {{5,3,8},{1,3,6},{2,5,4},{2,3,5},{4,5,6},{3,4,3},{2,4,8},{1,2,2},{1,4,5},{5,6,2}};
        System.out.println(solution.miniSpanningTree(6 ,10 ,cost));
    }

    /**
     * prim
     * @param n
     * @param m
     * @param cost
     * @return
     */
    public int miniSpanningTree (int n, int m, int[][] cost) {
        // write code here

        // 存储所有的点
        HashSet<Integer> pointList = new HashSet<>();
        // 存储所有的边
        List<int[]> edgeList = new ArrayList<>();

        // 定位出距离两个点之间距离最小的边
        Arrays.sort(cost, (o1, o2) -> o1[2] - o2[2]);

        // 把所有的边加入到结构集中
        for (int[] ints : cost) {
            edgeList.add(ints);
        }

        int res = edgeList.get(0)[2];
        // 先把两边距离最小的两天边加入到点集合中
        pointList.add(edgeList.get(0)[0]);
        pointList.add(edgeList.get(0)[1]);

        while (true){
            // 一直遍历找，找到于两条边相邻的点
            for (int i= 1; i < edgeList.size(); i++){
                // 找到与两条边相邻的顶点,并且这个边还是与一边相邻，另一边没有出现过
                if ((pointList.contains(edgeList.get(i)[0]) && !pointList.contains(edgeList.get(i)[1])) ||
                        ((pointList.contains(edgeList.get(i)[1])) && !pointList.contains(edgeList.get(i)[0]))){
                    res += edgeList.get(i)[2];
                    // 把找到的这条边加入到结果集中
                    pointList.add(edgeList.get(i)[0]);
                    pointList.add(edgeList.get(i)[1]);
                    // 删除这个边
                    edgeList.remove(edgeList.get(i));
                    break;
                }
            }
            // 全部村子都访问过了
            if (pointList.size() == n){
                break;
            }
        }
        return res;
    }

}