堆排序 Python 程式


在本文中,我們將瞭解下面給出的問題陳述的解決方案。

問題陳述 − 給定一個數組,我們需要使用堆排序的概念對其進行排序。

在此,我們將最大元素放在末尾。重複此過程,直至陣列排序。

現在,讓我們在下面的實現中觀察解決方案 −

示例

 演示

# heapify
def heapify(arr, n, i):
   largest = i # largest value
   l = 2 * i + 1 # left
   r = 2 * i + 2 # right
   # if left child exists
   if l < n and arr[i] < arr[l]:
      largest = l
   # if right child exits
   if r < n and arr[largest] < arr[r]:
      largest = r
   # root
   if largest != i:
      arr[i],arr[largest] = arr[largest],arr[i] # swap
      # root.
      heapify(arr, n, largest)
# sort
def heapSort(arr):
   n = len(arr)
   # maxheap
   for i in range(n, -1, -1):
      heapify(arr, n, i)
   # element extraction
   for i in range(n-1, 0, -1):
      arr[i], arr[0] = arr[0], arr[i] # swap
      heapify(arr, i, 0)
# main
arr = [2,5,3,8,6,5,4,7]
heapSort(arr)
n = len(arr)
print ("Sorted array is")
for i in range(n):
   print (arr[i],end=" ")

輸出

Sorted array is
2 3 4 5 5 6 7 8

所有變數都在區域性作用域中宣告,並且上圖中可以看到它們的引用。

結論

在本文中,我們學習瞭如何編寫 Python 程式進行堆排序

更新於: 20-Dec-2019

1K+ 瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

開始
廣告
© . All rights reserved.