前言
堆也是一种数据结构,是用数组来表示的完全二叉树
堆排序是利用堆这种数据结构来对数组进行排序的排序算法,其时间复杂度可以达到 O(n*logn)
(堆和堆排序真的很抽象,在写这篇文章的时候都感觉好难写啊,有写的不好的地方请指出来╥﹏╥)
手撕算法 - 排序系列
手撕面试题算法<排序>(1)—— 冒泡排序及其优化实现
手撕面试题算法<排序>(2)—— 选择排序
手撕面试题算法<排序>(3)—— 插入排序及其优化实现
手撕面试题算法<排序>(3.5)—— 希尔排序
手撕面试题算法<排序>(4)—— 归并排序
手撕面试题算法<排序>(5)—— 快速排序以及快排为什么快
手撕面试题算法<排序>(6)—— 堆 & 堆排序
手撕面试题算法<排序>(7)—— 箱排序 & 基数排序
源码
导航
堆
堆是一种用数组来表示的完全二叉树的数据结构,它有如下特点:
- 堆通常分为大根堆或小根堆两种
- 如果是大根堆,则二叉树中的每一个节点都会大于其子节点,其根节点为堆中的最大值
- 如果是小根堆则反之
- 假设堆的一个节点在数组中的索引为
i
,则其子节点索引为i*2+1
和i*2+2
- 例:根节点在数组中用arr[0]表示,其子节点是arr[1],arr[2],而arr[1]的子节点是arr[3],arr[4];arr[2]的子节点是arr[5],arr[6]。。。
堆排序
从堆的性质中,我们可以想到:如果我们可以一直从一个堆中拿出它的最大值 / 最小值,那不就可以用堆来排序了吗,这就是堆排序
通常我们在顺序排序数组的时候会构建大根堆,至于为什么,下面会讲到
如何堆排
- 将传入的数组初始化成大根堆
- 定义
j = arr.length-1
指向数组的末尾 - 将堆顶元素与
arr[j]
交换后,--j
指向数组的次末尾 - 根据区间[0, j]重构大根堆,使得堆顶的最大值是区间[0, j]的最大值
- 重复以上操作,直到数组顺序
大根堆的初始化
我们首先要将传入数组初始化为大根堆,根据大根堆的定义,二叉树中的每一个节点都会大于其子节点,其根节点为堆中的最大值
而堆(完全二叉树)的最后一个非叶子节点在数组中的索引是(arr.length/2)-1
:
- 我们定义
i = (arr.length/2)-1
- 从节点
arr[i]
的子树中搜索一个最大值 - 令最大值节点和
arr[i]
的值互换 - 让
i--
,重复以上直到i<0
,大根堆就初始化完毕了
大根堆的调整
将大根堆的堆顶替换后,若堆顶不是当前堆中的最大值,则需要进行一次调整
我们只需要搜索根节点的子树,从中获取到根节点的最大值,互换即可
大根堆的初始化和调整可以写成一个方法
重复以上步骤,就可以得到一条顺序序列而不需要创建新数组