堆的原理以及实现
本文是在好友哆啦奇梦的耐心指导下完成的,在此表示感谢!
-
大顶堆:
所有的父节点的值都比孩子节点大,叶子节点值最小。root 根节点是第一个节点值最大 -
小顶堆:
和大顶堆相反,所有父节点值,都小于子节点值,root 根节点是 第一个节点值最小
根据完全二叉树的性质,若将堆中的数据至顶向下,从左向右的存在一个一维数组里面,
则父节点的位置索引总是该节点位置索引减1再除2取整的结果。 p = (c-1) // 2
下面的图片来自这里
堆的列表索引对应的形状如下:
维护一个堆主要有两个操作:上浮和下沉:
如果是维护一个最小堆,上浮操作如下,子节点找他的父节点,如果子节点小于父节点,那么我们就交换父子节点的值
同上,如果是维护一个最小堆,我们的父节点比子节点大,就需要下沉。下沉的操作如下:
实现的代码如下:
from typing import List
"""
父节点与子节点的索引位置关系,p父节点,c子节点
p = (c-1)//2
c1 = 2*p+1
c2 = 2*p+2
本程序维护的是一个最大堆
"""
#对某一个父节点进行下沉,用于删除堆顶元数 arr数组;p节点;arr_len:数组长度
def downAdjust(arr, p, arr_len):
temp = arr[p]#找到要下沉的父节点的值
c = 2*p + 1 #计算出其左子节点的索引
#最大堆降序排序
while c <= arr_len:#当索引的位置在数组内的时候
if (c+1) <= arr_len and arr[c] > arr[c+1]:#当前儿子比另一个儿子大
c += 1#找到父节点的两个儿子中的较小的那一个
if temp <= arr