python堆排序输出下标的方法是什么

共计 598 个字符，预计需要花费 2 分钟才能阅读完成。

在 Python 中，可以使用 heapq 模块来实现堆排序，并输出元素的下标。

下面是一个示例代码：

import heapq

def heap_sort_with_index(arr):
    heap = []
    for i, num in enumerate(arr):
        heapq.heappush(heap, (num, i))
    
    sorted_arr = []
    while heap:
        sorted_arr.append(heapq.heappop(heap)[1])
    
    return sorted_arr

在上面的代码中，我们使用 heapq.heappush()将元素 (num, i) 加入到堆中，其中 num 是数组中的元素，i 是该元素的下标。这样，heap 中的每个元素都是一个元组，元组的第一个元素是数组中的元素，第二个元素是该元素的下标。

然后，我们使用 heapq.heappop()从堆中弹出元素，并将其下标添加到 sorted_arr 中。最终，返回 sorted_arr，即按从小到大排序的元素下标列表。

示例使用：

arr = [4, 2, 7, 1, 9, 5]
sorted_indexes = heap_sort_with_index(arr)
print(sorted_indexes)  # 输出 [3, 1, 0, 5, 2, 4]

上述示例中，原始数组 arr 为[4, 2, 7, 1, 9, 5]，经过堆排序后，得到的元素下标列表为[3, 1, 0, 5, 2, 4]。

丸趣 TV 网 – 提供最优质的资源集合！