中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

python堆排序輸出下標的方法是什么

小億
89
2023-10-22 02:53:26
欄目: 編程語言

在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]。

0
盐边县| 晋城| 河曲县| 大渡口区| 石首市| 仁化县| 邵武市| 嵩明县| 蒙山县| 巫溪县| 海伦市| 黑水县| 牙克石市| 西宁市| 蕉岭县| 柞水县| 区。| 拜泉县| 肇庆市| 都匀市| 外汇| 石景山区| 普陀区| 濉溪县| 义马市| 华容县| 泰宁县| 闵行区| 屯留县| 加查县| 弥渡县| 龙山县| 稷山县| 静宁县| 绩溪县| 黎城县| 安陆市| 大兴区| 衡阳县| 泾源县| 靖江市|