堆是一种特殊的树,数只要满足这两点,它就是堆:
- 堆是一个完全二叉树;
完全二叉树要求,除最后一层,其它层的节点个数都是满的,最后一层的节点都靠左排列。 - 堆中每一个节点的值都必须大于等于(或小于等于)其子树中每个节点的值。
也可以说,堆中每个节点的值都大于等于(或者小于等于)其左右子节点的值。
对于每个节点的值都大于等于子树中每个节点值的堆,我们叫作“大顶堆”。对于每个节点的值都小于等于子树中每个节点值的堆,我们叫作“小顶堆”。
图中1,2是大顶堆,3是小顶堆。第4个不是堆。从图中还可以看出来,对于同一组数据,我们可以构建多种不同形态的堆。
可以用一个数组存储堆:
从图中可以看出,数组中下标为 i 的节点的左子节点,就是下标为i2的节点,右子节点就是下标为i2+1的节点,父节点就是下标为i/2的节点。
往堆中插入一个元素
往堆中插入一个元素后,需要继续满足堆的两个特性。让其重新满足堆的特性的过程叫作堆化。
堆化的方法包括从下往上和从上往下,先学习从下往上的方法:
我们让新插入的节点与父节点对比大小,如果不满足子节点小于等于父节点的大小关系,我们就互换两个节点。一直重复这个过程,直到父子节点之间满足堆中的定义。
删除堆顶元素
由堆的定义的第二条,我们可以发现,堆顶元素存储的就是堆中数据的最大值或者最小值。
运用从上往下的堆化方法进行删除操作。即将最后一个节点放到堆顶,然后利用同样的父子节点对比方法。对于不满足父子节点大小关系的,互换两个节点,并且重复进行这个过程,直到父子节点之间满足大小关系为止。
我们知道,一个包含n个节点的完全二叉树,树的高度不会超过log以2为底n的对数。堆化的过程就是顺着节点所在的路径比较交换的,所以往堆中插入一个元素和删除堆顶元素的时间复杂度都是O(logn)。
堆排序
堆排序的过程主要分为两个步骤:建堆和排序。
建堆分解步骤图如下所示:
因为叶子节点往下堆化只能自己跟自己比较,所以我们直接从第一个非叶子节点开始,依次堆化就行了。
建堆的时间复杂度是O(n)。
建堆结束之后,数组中的数据已经是按照大顶堆的特性来组织的,这时我们进行排序操作,这个过程有点类似“删除堆顶元素的操作”,当堆顶元素移除之后,我们把下标为n的元素放到堆顶,然后再通过堆化的方法,将剩下的n-1个元素重新构建成堆。堆化完成之后,我们再取堆顶的元素,放到下标为n-1的位置,一直重复这个过程,直到最后堆中只剩下标为1的一个元素,排序工作就完成了。
排序过程的时间复杂度是O(nlogn),所以堆排序整体的时间复杂度是O(nlogn)。
整个对排序的过程,都只需要极个别临时存储空间,所以堆排序是原地排序算法。因为在堆排序的过程中,存在将堆的最后一个节点与堆顶节点互换的操作,这就有可能改变值相同数据的原始相对顺序,所以堆排序不是稳定的排序算法。
堆的应用:优先级队列,求Top K和求中位数。