数据结构与算法学习-二叉堆及堆排序

1. 二叉堆定义

(1)堆是一个完全二叉树

(2)堆中每一个节点的值都必须大于等于(或小于等于)其子树中每个节点的值。

第一点,堆必须是一个完全二叉树。,除最后一层,其他层的节点个数都是满的,最后一层的节点都靠左排列。

第二点,堆中的每个节点的值必须大于等于(或者小于等于)其子树中每个节点的值。也可以理解为,堆中每个节点的值都大于等于(或者小于等于)其左右子节点的值。

这里我们借助于堆这种数据结构实现的排序算法,就叫作堆排序。这种排序方法的时间复杂度非常稳定,是O(nlogn),并且它还是原地排序算法。

对于每个节点的值都大于等于子树中每个节点值的堆,叫作“大顶堆”

big_heap.png

对于每个节点的值都小于等于子树中每个节点值的堆,叫作“小顶堆”

small_heap.png

2. 堆的操作

堆的操作基本就是 3 种,掺入、删除和排序,下面就讲一下这几种操作的过程以及代码实现。

完全二叉树比较适合用数组来存储。用数组来存储完全二叉树是非常节省存储空间,不需要存储左右子节点的指针,单纯地通过数组的下标,就可以找到一个节点的左右子节点和父节点。

左子节点 = 父节点 * 2 + 1

右子节点 = 父节点 * 2 + 2

下面几个过程均以大顶堆为例,小顶堆同样道理,只是大小相反。

2.1 插入节点

插入一个节点,需要将节点放到数组的尾部,然后执行向上堆化过程,这里成为【上浮】,保证堆顶部的元素最大,对每个父节点来说,均大于等于子节点。

public class Heap {

    /**
     * 堆的数组
     */
    private int[] arr;
    /**
     * 堆的最大数据个数
     */
    private int max;
    /**
     * 堆中数据个数
     */
    private int size;

    public Heap(int capacity) {
        arr = new int[capacity];
        max = capacity;
        size = 0;
    }

    // 插入节点
    public void insert(int value) {
        if (size == max) {
            return;
        }
        arr[size] = value;
        int childIndex = size;
        int parentIndex = (childIndex - 1) / 2;
        while (childIndex > 0 && value > arr[parentIndex]) {
            arr[childIndex] = arr[parentIndex];
            childIndex = parentIndex;
            parentIndex = (parentIndex - 1) / 2;
        }
        arr[childIndex] = value;
        size++;
    }

2.2 删除节点

删除节点,对于堆来说,删除一般是从堆顶删除,对于大顶堆,就是删除最大的元素,那么删除之后就需要再从堆中找到最大的元素,放到顶部。假设最后一个元素最大,把它放在顶部,然后执行向下堆化过程,这里称为【下沉】,执行这个过程之后,就将顶部元素下沉到某一个位置,每次下沉的条件是子元素的值比父元素大,需要完成交换,所以执行之后,就可以保证父节点的值大于等于子节点的值。

    public int removeMax() {
        if (arr == null || size == 0) {
            return -1;
        }
        size--;
        int result = arr[0];
        arr[0] = arr[size];
        downAdjust(arr, size, 0);
        return result;
    }

    /**
     * 下沉
     *
     * @param arr  数组
     * @param size 数组大小
     * @param i    索引
     */
    private static void downAdjust(int[] arr, int size, int i) {
        int temp = arr[i];
        int parentIndex = i;
        int childIndex = 2 * parentIndex + 1;
        while (childIndex < size) {
            // 如果有右孩子,且右孩子大于左孩子的值,则定位到右孩子
            if (childIndex + 1 < size && arr[childIndex] < arr[childIndex + 1]) {
                childIndex++;
            }
            if (temp >= arr[childIndex]) {
                break;
            }
            arr[parentIndex] = arr[childIndex];
            parentIndex = childIndex;
            childIndex = 2 * childIndex + 1;
        }
        arr[parentIndex] = temp;
    }

2.3 排序

堆排序需要两个过程:建堆和

建堆:就是让所有非叶子节点依次下沉,从倒数第二排非叶子元素开始执行,一直循环到第一个元素,这样就保证所有父节点大于等于子节点,堆顶元素是最大的。

排序:建堆之后,由于堆顶元素最大,那么可以将最大元素和最后一个元素交换,这样最大值就确定了;然后将剩下的元素执行堆化,也就是将刚刚交换的元素下沉操作,这个过程之后,堆顶元素又是最大的,然后重复这个过程,就能将堆中元素完成从小到大的排序。

时间复杂度:

(1)每个堆化过程时间复杂度是 O(logn),因为树的高度是 logn,建堆总共次数是 n/2 * logn,所以建堆过程时间复杂度是 O(nlogn)

(2)排序过程:每个下沉过程时间复杂度是O(logn),一共执行 n - 1 次,所以排序过程就是 O(nlogn)

最终堆排序过程就是 O(nlogn)

    /**
     * 构建堆
     *
     * @param arr    数组
     * @param length 长度
     */
    private static void buildHeap(int[] arr, int length) {
        for (int i = (length - 2) / 2; i >= 0; i--) {
            downAdjust(arr, length, i);
        }
    }

    /**
     * 堆排序
     *
     * @param arr 数组
     * @param n   n为数组长度
     */
    public static void sort(int[] arr, int n) {
        if (arr == null) {
            return;
        }
        buildHeap(arr, n);
        int k = n - 1;
        while (k > 0) {
            // 交换堆顶元素和最后一个元素
            int temp = arr[0];
            arr[0] = arr[k];
            arr[k] = temp;
            k--;
            downAdjust(arr, k, 0);
        }
    }

参考

漫画:什么是二叉堆

代码地址

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,992评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,212评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,535评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,197评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,310评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,383评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,409评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,191评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,621评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,910评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,084评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,763评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,403评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,083评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,318评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,946评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,967评论 2 351

推荐阅读更多精彩内容