排序算法(七):堆排序

def adjust_heap(arr, i, size):
    lchild = 2 * i + 1
    rchild = 2 * i + 2
    max = i
    if i < size / 2:
        if lchild < size and arr[lchild] > arr[max]:
            max = lchild
        if rchild < size and arr[rchild] > arr[max]:
            max = rchild
        if max != i:
            arr[max], arr[i] = arr[i], arr[max]
            adjust_heap(arr, max, size)

def build_heap(arr, size):
    for i in range(0, round(size / 2))[::-1]:
        adjust_heap(arr, i, size)

def heap_sort(arr):
    size = len(arr)
    build_heap(arr, size)
    for i in range(0, size)[::-1]:
        arr[0], arr[i] = arr[i], arr[0]
        adjust_heap(arr, 0, i)
    return arr
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容