堆排序 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+ 浏览

开启你的 职业生涯

通过完成课程获得认证

开始
广告