使用python写一个堆排序
时间: 2023-03-23 11:01:15 浏览: 128
好的,下面是使用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[0], arr[i] = arr[i], arr[0]
heapify(arr, i, 0)
return arr
def heapify(arr, n, i):
largest = i
l = 2 * i + 1
r = 2 * i + 2
if l < n and arr[l] > arr[largest]:
largest = l
if r < n and arr[r] > arr[largest]:
largest = r
if largest != i:
arr[i], arr[largest] = arr[largest], arr[i]
heapify(arr, n, largest)
```
这个代码实现了堆排序,其中`heap_sort`函数接收一个列表参数,返回一个排好序的列表。在`heap_sort`函数中,首先通过`heapify`函数构建了一个大根堆,然后进行排序。`heapify`函数用于将一个无序的列表调整为一个大根堆。在`heapify`函数中,首先找出左右子节点中较大的一个,然后将父节点与子节点进行比较,如果父节点小于子节点,则交换两者的位置,并递归进行调整,使得最终的结果是一个大根堆。
阅读全文