堆排序是一种基于二叉堆数据结构的排序算法,它的基本思想是将待排序的数据转换成一个二叉堆,然后不断地将堆顶元素(即最大或最小元素)取出来,放到已排序部分的末尾,再重新调整剩余未排序部分的堆,直到全部排序完成。
堆排序的主要步骤包括:
- 构建二叉堆:将待排序数组看作一个完全二叉树,从最后一个非叶子节点开始,从下往上依次进行“下沉”操作,使得每个节点都满足堆的性质(即父节点的值大于或小于它的子节点的值)。
- 取出堆顶元素:将堆顶元素取出,放到已排序部分的末尾。
- 调整堆:将剩余未排序部分重新调整成一个堆,重复步骤2和步骤3,直到全部元素排序完成。
下面是用Python实现堆排序算法的代码:
def heapify(arr, n, i):
"""
将以i为根节点的子树调整成一个堆
"""
largest = i
left = 2 * i + 1
right = 2 * i + 2
if left < n and arr[left] > arr[largest]:
largest = left
if right < n and arr[right] > arr[largest]:
largest = right
if largest != i:
arr[i], arr[largest] = arr[largest], arr[i]
heapify(arr, n, largest)
def heap_sort(arr):
n = len(arr)
# 构建最大堆
for i in range(n // 2 - 1, -1, -1):
heapify(arr, n, i)
# 依次取出堆顶元素,调整堆
for i in range(n - 1, 0, -1):
arr[0], arr[i] = arr[i], arr[0]
heapify(arr, i, 0)
return arr
在上面的代码中,heapify函数用来将以i为根节点的子树调整成一个堆,heap_sort函数则用来实现堆排序。在heap_sort函数中,首先构建最大堆,然后依次取出堆顶元素,调整剩余未排序部分的堆,最终得到排序好的数组。
下面是一个示例,展示如何使用堆排序算法对一个数组进行排序:
arr = [12, 11, 13, 5, 6, 7]
sorted_arr = heap_sort(arr)
print(sorted_arr) # [5, 6, 7, 11, 12, 13]