共計(jì) 598 個(gè)字符,預(yù)計(jì)需要花費(fèi) 2 分鐘才能閱讀完成。
在 Python 中,可以使用 heapq 模塊來(lái)實(shí)現(xiàn)堆排序,并輸出元素的下標(biāo)。
下面是一個(gè)示例代碼:
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 是數(shù)組中的元素,i 是該元素的下標(biāo)。這樣,heap 中的每個(gè)元素都是一個(gè)元組,元組的第一個(gè)元素是數(shù)組中的元素,第二個(gè)元素是該元素的下標(biāo)。
然后,我們使用 heapq.heappop()從堆中彈出元素,并將其下標(biāo)添加到 sorted_arr 中。最終,返回 sorted_arr,即按從小到大排序的元素下標(biāo)列表。
示例使用:
arr = [4, 2, 7, 1, 9, 5]
sorted_indexes = heap_sort_with_index(arr)
print(sorted_indexes) # 輸出 [3, 1, 0, 5, 2, 4]
上述示例中,原始數(shù)組 arr 為[4, 2, 7, 1, 9, 5],經(jīng)過(guò)堆排序后,得到的元素下標(biāo)列表為[3, 1, 0, 5, 2, 4]。
丸趣 TV 網(wǎng) – 提供最優(yōu)質(zhì)的資源集合!
正文完