python实现堆排序，常用于topn问题，时间复杂度为nlogn

1. 实现堆的构建：构建一个最大堆或最小堆，将数组转换为符合堆结构的形式

def build_heap(arr):
    n = len(arr)
    # 从最后一个非叶子节点开始，向上调整每个子树使其满足堆的性质
    for i in range(n // 2 - 1, -1, -1):
        heapify(arr, n, i)


def heapify(arr, n, i):
    largest = i  # 当前节点的索引
    left = 2 * i + 1  # 左子节点的索引
    right = 2 * i + 2  # 右子节点的索引

    # 比较左子节点与当前节点的值，更新最大值索引
    if left < n and arr[left] > arr[largest]:
        largest = left

    # 比较右子节点与当前节点的值，更新最大值索引
    if right < n and arr[right] > arr[largest]:
        largest = right

    # 如果最大值索引不是当前节点，则交换当前节点与最大值节点的位置，并递归调整子树
    if largest != i:
        arr[i], arr[largest] = arr[largest], arr[i]
        heapify(arr, n, largest)

2. 实现堆排序函数：使用构建好的堆进行排序。

def heap_sort(arr):
    n = len(arr)

    # 构建最大堆
    build_heap(arr)

    # 从最后一个节点开始，依次将根节点与当前未排序部分的最后一个元素交换，并调整堆
    for i in range(n - 1, 0, -1):
        arr[0], arr[i] = arr[i], arr[0]  # 将根节点与当前未排序部分的最后一个元素交换
        heapify(arr, i, 0)  # 调整堆，将最大值放到根节点


if __name__ == "__main__":
    q = [1, 4, 2, 3, 7, 5, 5]
    heap_sort(q)
    print(q)