我按照 clrs 的书研究了算法。我正在尝试在 python 中制作堆排序。但它给了我一个错误,即 r 掉在索引的一侧,但我不知道为什么。
def Max_Heapify(A,i,size_of_array):
l = 2*i
r = l + 1
if l <= size_of_array and A[l] > A[i]:
largest = l
else:
largest = i
if r <= size_of_array and A[r] > A[largest]:
largest = r
if i != largest:
A[i], A[largest] = A[largest], A[i]
Max_Heapify(A,largest,size_of_array)
def Build_Max_Heap(A,size_of_array):
for i in range((math.floor(size_of_array/2)) - 1 , 0 ,-1):
Max_Heapify(A,i,size_of_array)
def Heapsort(A,size_of_array):
Build_Max_Heap(A,size_of_array)
for i in range(size_of_array - 1 ,0 ,-1):
A[0],A[i] = A[i],A[0]
size_of_array = size_of_array - 1
Max_Heapify(A,0,size_of_array)
在大多数编程语言中,数组的大小大于最后一个索引。例如,下面的数组:A = [1, 2, 3]
,它的大小是 3,但最后一个元素的索引是 2(A[3]
应该返回它不在索引中)。您正在验证 r 是否小于或等于数组大小,因此当它相等时,它大于最后一个索引。您的验证应该是:
if r < size_of_array