以大顶堆为例,首先贴出代码
def adjustHeap(arr,i,n):
#基本思想是和自己孩子节点中大的那个相比较,如果孩子节点大,则和该节点交换,然后把大的那个孩子节点处作为根节点继续比较下去。
temp = arr[i] #首先把当前节点保存下来以免后面覆盖掉j= 2*i + 1 #左孩子是2i+1
while j < n: #孩子节点超过最大值就不需要比较了,说明已经没有孩子节点了
if j+1 < n and arr[j] < arr[j + 1]: #如果右孩子大,就和右孩子比
j+= 1
if temp < arr[j]: #如果孩子节点大,就用孩子节点覆盖根节点,因为采用的类似填坑方法,所以把此时孩子节点位置记录下来,注意这里每次都是跟最开始的根节点比,这样才能找到这个根节点在以它为根的子树上的位置
arr[i] =arr[j]
i = j
else:
break
j = 2*j+1 #然后把孩子节点当做根节点,继续循环
print(arr)
arr[i] = temp #把最开始的根节点填到最后的坑里
print(arr)
#建立初始堆,建立完后arr[0]为最大
#从最后一个非叶节点开始逐个往前比较,因为在比较时是和 自己的孩子比较,叶节点已经没有孩子所以不需比较
for i in range(len(arr)//2 - 1,-1,-1):print(i)
adjustHeap(arr,i,len(arr))
print('final\n',arr)
#每次把最大的放到最后的有序区,然后再对前面的无序区(上次堆的长度-1)重新建立堆
#现在从根节点开始调整
j = len(arr) - 1while j > 0:
print('j =',j)
t = arr[j]
arr[j] = arr[0]
arr[0] = t
adjustHeap(arr,0,j)
j -= 1
第一步:建立初始堆
第二步:把最大值,此时为arr[0]放到最后
以后开始循环,