一百亿个数据找出其中的一千个最大的——堆排序

# 堆排序函数
def heap_sort(arr):
    n = len(arr)

    # 构建最大堆
    for i in range(n // 2 - 1, -1, -1):
        # 对每个非叶子节点进行堆调整
        heapify(arr, n, i)

    # 从最大堆中取出k个最大元素
    result = []
    for i in range(n - 1, n - 1001, -1):
        # 将堆顶元素与最后一个元素交换位置
        arr[0], arr[i] = arr[i], arr[0]
        # 将堆顶元素添加到结果列表中
        result.append(arr[i])
        # 对交换后的堆顶元素进行堆调整
        heapify(arr, i, 0)

    return result

# 调整堆
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)

# 测试代码
data = [10000000000个数据]
result = heap_sort(data)
print("一千个最大元素：", result)