堆排序的python函数代码

以下是堆排序的 Python 函数代码：

def heapify(arr, n, i):
    largest = i  # 初始化根节点为最大值
    l = 2 * i + 1    # 左子节点
    r = 2 * i + 2    # 右子节点
 
    # 如果左子节点大于根节点，则更新最大值
    if l < n and arr[i] < arr[l]:
        largest = l
 
    # 如果右子节点大于根节点，则更新最大值
    if r < n and arr[largest] < arr[r]:
        largest = r
 
    # 如果最大值不是根节点，则交换它们，并递归地堆化子树
    if largest != i:
        arr[i], arr[largest] = arr[largest], arr[i]
 
        heapify(arr, n, largest)
 
def heapSort(arr):
    n = len(arr)
 
    # 构建最大堆
    for i in range(n, -1, -1):
        heapify(arr, n, i)
 
    # 依次取出最大值，放到末尾，并堆化剩余部分
    for i in range(n-1, 0, -1):
        arr[i], arr[0] = arr[0], arr[i]
        heapify(arr, i, 0)

这个函数实现了一个堆排序算法，通过构建最大堆来实现排序。首先，我们从最后一个非叶子节点开始构建最大堆，然后依次取出最大值并将其放到末尾，同时堆化剩余部分，直到整个数组都被排序完成。