生成一份python的堆排序
时间: 2023-10-12 16:09:33 浏览: 54
以下是Python的堆排序实现:
```python
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[i], arr[0] = arr[0], arr[i]
heapify(arr, i, 0)
def heapify(arr, n, 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)
```
使用示例:
```python
arr = [12, 11, 13, 5, 6, 7]
heap_sort(arr)
print(arr)
```
输出结果:
```
[5, 6, 7, 11, 12, 13]
```