概念
堆是按照一定规则顺序存储的完全二叉树
(二叉树是每个节点最多有两个子树的树结构),其中可以分为大根堆和小根堆。
* 大根堆: 每个父结点的关键字都大于其子节点的关键字(如果有子结点的话)
* 小根堆: 每个父结点的关键字都小于其子节点的关键字(如果有子结点的话)
举例来说,有个数组[A0, A1, A2, A3…An]假设父结点为Ai(其中i=1,2,…,n/2向下取整), 由二叉树的排列规律可知其左子结点为A(2 * i + 1), 右子结点为A(2 * i + 2)因此大根堆
必须满足:Ai >= A(2 * i + 1)且Ai >= A(2 * i + 2), 小根堆
必须满足Ai <= A(2 * i + 1)且Ai <= A(2 * i + 2)
假如有个数列[3, 8, 15, 31, 25],以二叉树的形式表示如下图:
首先需要按照堆的规则去构造初始堆(即构建一个完全二叉树,保证所有的父结点都比它的孩子结点数值大),构造规则为不同行从下往上、相同行从右往左
分别比较每一个子二叉树的结点,如果父结点的关键字不是最大值,让子节点中最大的值与父节点进行交换。如此遍历一遍后可以得到一个初始堆。
注意: 当第一级二叉树子结点值大于父结点的值进行交换后,交换后新的子结点作为父结点的第二级子二叉树中如果依然不是最大值则需要再次与其最大子结点进行交换。例:有个数组[5,3,1,8,7]在调整后变为[5,8,1,3,7]->[8,5,1,3,7],然而此时由于在(5,3,7)这个子二叉树中父结点5不是最大值,因此需要再次向下比较一次进行交换
在初始堆的基础上交换数组中被调整的首尾元素,并让数组重新调整为大根堆(这次调整忽略移到末尾的元素),采用递归的方式就可以依次把所有元素按升序排列到数组中了。
OC语言实例
给可变数组创建一个分类来存放该堆排序方法。
- (NSMutableArray *)heapSort
{