【算法】堆排序算法的讲解和代码实践

思路

学习堆排序之前,先简单介绍一下什么是堆。
但是要了解堆,还得先连接二叉树。
二叉树,就是每个节点最多有两个子节点的树形的数据结构。二叉树又有完全二叉树、满二叉树等等,细的就不讲了,可以搜一下资料,还是很多讲这个的。
二叉堆,它是属于完全二叉树,同时它又满足堆的性质(子节点的值总是小于它的父节点(也可以是大于))。也就是说,根节点肯定是最大(也可能是最小)的那个元素。
堆排序就是根据这个特点,把数组看成是一个二叉堆,每次计算出一个最大或最小值然后进行排序的。
那么完全二叉树它怎么跟数组关联上呢?其实完全二叉树它有一个特点,就是它完全可以用数组来表示。详细的这里就不讲了,但是有几点需要记住。

  1. 最后一个非叶子节点下标 = 数组长度 ÷ 2 - 1;
  2. 非叶子节点的左子节点下标 = 该节点下标 × 2 + 1;
  3. 非叶子节点的右子节点下标 = 该节点下标 × 2 + 2。

知道这三点之后,我们就可以根据完全二叉树的性质,把它编程一个二叉堆,然后进行排序了。
关于满二叉树和完全二叉树我还是简单画个图吧,帮助理解。


满二叉树

这种就叫满二叉树,左右完全平衡,非叶子节点都有两个子节点,叶子节点都在一层。
完全二叉树就是说,满二叉树的基础上,比如说要加一个节点,那么只能先在D下面加,然后依次往后。要删除一个节点,只能从右下角开始,先把G删掉,再删F,再删E这样。


完全二叉树

那么完全二叉树跟数组的关系呢?我们加一个数组看一下:
完全二叉树和数组的关系

就可以看到了,比如最后一个非叶子节点是C,下标为2,数组长度是6,6 ÷ 2 - 1 = 2,就印证了上面的公式。

OK二叉树讲的有点多了,下面进入整体,开始将堆排序。

讲解

依然是有这样一个数组:


image.png

那么我们先把它看成一个完全二叉树,开始把它构建成一个最大堆,第一步,找到最后一个非叶子节点,用上面的公式,数组长度为8,最后一个非叶子节点就是 8 ÷ 2 - 1 = 3,那么最后一个非叶子节点就找到了。最后一个非叶子节点前面的元素,都是非叶子节点,我们用蓝色标记出来:


image.png

找到非叶子节点之后,我们从最后一个非叶子节点开始向前遍历所有的非叶子节点。
先看最后一个非叶子节点 -15 ,它是否小于它的两个(也可能是一个)叶子节点。根据公式计算,计算出它的两个叶子下标分别为:3 × 2 + 1 = 7,3 × 2 + 2 = 8,下标为8的元素很显然不存在,那么 -15 只需要跟22对比,明显 -15 小于 22,交换两个元素的位置:
image.png

然后遍历 14,14 的两个子节点下标为 2 × 2 + 1 = 5,2 × 2 + 2 = 6,所以 14 需要与 324 和 1 进行对比,14 比 324 小,交换位置:


image.png

324 与 1 比,324 大,不再交换。
下面轮到 -3。-3 的两个子节点下标为 1 × 2 + 1 = 3,1 × 2 + 2 = 4,-3 与 22、0 对比。-3 小于 22,交换位置:
image.png

22 比 0 大,不再交换。
下一步,26,两个子节点为 1、 2,26 与 22、324 对比。比 22 大,不交换,比 324 小,交换:
image.png

这样就形成了一个最大堆,每一个非叶子节点都大于它自己的叶子节点。最大值就在根节点,也就是我们数组下标为 0 的位置。接下来,我们把最大的数与数组最后一个元素交换位置:
image.png

这样最大的数就跑到了数组的末尾,第一轮的排序也就结束了。
那么第二轮,我们将已排好序的大数排除到构建最大堆的范围之外,只用前面7个数去构件最大堆,构建完成之后,再把最大数挪到下标6的位置,以此类推,即可完成整个数组的排序。

实现

    @Test
    public void heapSort() {
        int[] nums = new int[]{26, -3, 14, -15, 0, 324, 1, 22};
        sort(nums);
        System.out.println(Arrays.toString(nums));
    }

    private void sort(int[] nums) {
        // 如果数组长度小于2,直接返回
        if (nums.length < 2) {
            return;
        }
        // 遍历
        for (int i = 0; i < nums.length; i++) {
            // 构建最大堆,nums.length - i 是为了排除掉以排好序的元素
            int lastParentNode = (nums.length - i) / 2 - 1;
            // 遍历所有非叶子节点
            for (int parentNode = lastParentNode; parentNode >= 0; parentNode--) {
                // 如果该节点小于它的左子节点,那么交换位置
                if (nums[parentNode] < nums[parentNode * 2 + 1]) {
                    swap(nums, parentNode, parentNode * 2 + 1);
                }
                // 如果该节点存在右子节点,并且该节点小于它的右子节点,那么交换位置
                if (parentNode * 2 + 2 < nums.length - i && nums[parentNode] < nums[parentNode * 2 + 2]) {
                    swap(nums, parentNode, parentNode * 2 + 2);
                }
            }
            // 构建最大堆完成,把最大的数与堆的最后一个元素交换位置
            swap(nums, 0, nums.length - i - 1);
        }
    }

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

推荐阅读更多精彩内容