学习笔记:最小生成树算法

一、普里姆(Prim)算法

①初始化新图仅包含原图中的任意一个顶点,不包含任何边。

②从原图中选择一条权值最小的边,该边满足有且仅有一个顶点在新图中。将该边加入新图。

③重复直至所有顶点都在新图中,新图即最小生成树。

二、克鲁斯卡尔(Kruskal)算法

①初始化新图包含原图中的所有顶点,不包含任何边。

②从小到大遍历原图中所有边,若边中的两个顶点在新图中不存在连通路径,则将其加入新图。

③遍历结束后,新图即最小生成树。

三、实现

import java.util.TreeMap;
import java.util.TreeSet;

public class MinimumSpanningTree {
    public static class Edge implements Comparable<Edge> {
        int u, v, w;

        public Edge(int u, int v, int w) {
            this.u = u;
            this.v = v;
            this.w = w;
        }

        @Override
        public int compareTo(Edge e) {
            if (w != e.w) return w - e.w;
            else if (u != e.u) return u - e.u;
            else return v - e.v;
        }
    }

    public static TreeSet<Edge> prim(TreeSet<Integer> V, TreeSet<Edge> E) {
        TreeSet<Edge> T = new TreeSet<>();
        V.remove(V.first());
        while (!V.isEmpty())
            for (Edge e : E)
                if (V.contains(e.u) != V.contains(e.v)) {
                    V.remove(e.u);
                    V.remove(e.v);
                    T.add(e);
                    E.remove(e);
                    break;
                }
        return T;
    }

    public static TreeSet<Edge> kruskal(TreeSet<Integer> V, TreeSet<Edge> E) {
        TreeSet<Edge> T = new TreeSet<>();
        TreeMap<Integer, Integer> comp = new TreeMap<>();
        for (Integer v : V)
            comp.put(v, comp.size());
        for (Edge e : E) {
            if (comp.get(e.u).equals(comp.get(e.v)))
                continue;
            for (Integer i : comp.keySet())
                if (i != e.u && comp.get(i).equals(comp.get(e.u)))
                    comp.put(i, comp.get(e.v));
            comp.put(e.u, comp.get(e.v));
            T.add(e);
        }
        return T;
    }
}

原文地址:https://www.cnblogs.com/arseneyao/p/8343344.html

时间: 2024-11-07 15:22:01

学习笔记:最小生成树算法的相关文章

算法学习笔记 KMP算法之 next 数组详解

最近回顾了下字符串匹配 KMP 算法,相对于朴素匹配算法,KMP算法核心改进就在于:待匹配串指针 i 不发生回溯,模式串指针 j 跳转到 next[j],即变为了 j = next[j]. 由此时间复杂度由朴素匹配的 O(m*n) 降到了 O(m+n), 其中模式串长度 m, 待匹配文本串长 n. 其中,比较难理解的地方就是 next 数组的求法.next 数组的含义:代表当前字符之前的字符串中,有多大长度的相同前缀后缀,也可看作有限状态自动机的状态,而且从自动机的角度反而更容易推导一些. "前

Machine Learning In Action 第二章学习笔记: kNN算法

本文主要记录<Machine Learning In Action>中第二章的内容.书中以两个具体实例来介绍kNN(k nearest neighbors),分别是: 约会对象预测 手写数字识别 通过“约会对象”功能,基本能够了解到kNN算法的工作原理.“手写数字识别”与“约会对象预测”使用完全一样的算法代码,仅仅是数据集有变化. 约会对象预测 1 约会对象预测功能需求 主人公“张三”喜欢结交新朋友.“系统A”上面注册了很多类似于“张三”的用户,大家都想结交心朋友.“张三”最开始通过自己筛选的

[算法学习笔记]排序算法——堆排序

堆排序 堆排序(heapsort)也是一种相对高效的排序方法,堆排序的时间复杂度为O(n lgn),同时堆排序使用了一种名为堆的数据结构进行管理. 二叉堆 二叉堆是一种特殊的堆,二叉堆是完全二叉树或者是近似完全二叉树.二叉堆满足堆特性:父节点的键值总是保持固定的序关系于任何一个子节点的键值,且每个节点的左子树和右子树都是一个二叉堆. 如上图显示,(a)是一个二叉堆(最大堆), (b)是这个二叉堆在数组中的存储形式. 通过给个一个节点的下标i, 很容易计算出其父节点,左右子节点的的下标,为了方便,

STL学习笔记(算法概述)

算法头文件 要运用C++标准程序库的算法,首先必须包含头文件<algorithm> 使用STL算法时,经常需要用到仿函数以及函数配接器.它们定义域<functional>头文件中. 算法的分类 可以按以下分类方式描述各个STL算法: 非变动性算法(nonmodifying algorithms) 变动性算法(modifying algorithms) 移除性算法(removing algorithms) 变序性算法(mutating algorithms) 排序算法(sorting

ios学习笔记---排序算法

排序算法 1.概念 所谓排序,就是使一串记录,按照其中的某个或某些关键字的大小,递增或递减的排列起来的操作.排序算法,就是如何使得记录按照要求排列的方法. 2.选择排序算法时常用的几个参照 a.稳定性 假定在带排序的记录序列中,存在多个具有相同关键字的记录,若经过排序,这些记录的相对次序保持不变,即在原序列中,ri = rj,且ri在rj之前,而在排序后的序列中,ri仍在rj之前,则称这种排序算法是稳定的:否则称为不稳定的. b.时间复杂度 c.空间复杂度 3.算法 冒泡排序 选择排序 插入排序

Java学习笔记——排序算法之进阶排序(堆排序与分治并归排序)

春蚕到死丝方尽,蜡炬成灰泪始干 --无题 这里介绍两个比较难的算法: 1.堆排序 2.分治并归排序 先说堆. 这里请大家先自行了解完全二叉树的数据结构. 堆是完全二叉树.大顶堆是在堆中,任意双亲值都大于(或等于)其孩子值,就称其为大顶堆. 堆排序的步骤: 1.把数组想象成一个堆.数组的index+1就是其对应在堆中的序号 2.调堆中各值的顺序,得到大顶堆 3.将堆首位值与堆末尾值交换,最大值排序完毕 4.将堆得大小减1,重复步骤2和步骤3,直到堆中只剩下一个元素.排序完毕 上代码: 1 publ

【网络流】网络流学习笔记Part2ISAP算法

说实话ISAP的文献真的不太好找= =而且介绍的没有太详细,不像SAP Dinic比较普及. ISAP其实是改进的SAP算法,要学ISAP就先去看一下SAP好了.(事实上很多人会把ISAP和SAP搞混了.尤其在国内,很多人会直接管ISAP叫SAP) SAP算法(即Edmonds-Karp算法): 不断进行BFS找增广路径,那么最多找V*E次就一定不存在增广路径了. 时间复杂度 O(V*E^2) ISAP算法: 通过维护距离标号使得寻找增广路径的过程被简化从而提高效率.距离标号可以使某个点到汇点s

OPENCV学习笔记15_算法设计中使用策略模式

Building a bug-free(无BUG) application is just the beginning. What you really want is an application that you and the programmers working with you(团队) will be able to easily adapt and evolve (修改和升级)as new requirements come in(随着新的需求进入,面临新的需求). Basical

Java学习笔记——排序算法之希尔排序(Shell Sort)

落日楼头,断鸿声里,江南游子.把吴钩看了,栏杆拍遍,无人会,登临意. --水龙吟·登建康赏心亭 希尔算法是希尔(D.L.Shell)于1959年提出的一种排序算法.是第一个时间复杂度突破O(n2)的算法之一. 其基础是插入排序. 上代码: 1 public class ShellSort { 2 3 public static void shellSort(int[] arr){ 4 5 int increment = arr.length; 6 int temp;//牌 7 int i; 8

学习笔记 ST算法

[引子]RMQ (Range Minimum/Maximum Query)问题: 对于长度为n的数列A,回答若干询问RMQ(A,i,j)(i,j<=n),返回数列A中下标在i,j里的最小(大)值,也就是说,RMQ问题是指求区间最值的问题. {方法} 1.朴素(即搜索),O(n)-O(qn) online. 2.线段树,O(n)-O(qlogn) online. 3.ST(实质是动态规划),O(nlogn)-O(q) online. ST算法(Sparse Table),以求最大值为例,设d[i,