Pycharm中堆排序算法代码
时间: 2024-06-22 16:02:32 浏览: 98
在PyCharm中编写堆排序算法,首先需要了解堆是一种特殊的树形数据结构,通常用于实现优先队列。堆排序是利用堆这种数据结构进行的一种排序方法,它的时间复杂度为O(n log n)。
以下是一个简单的Python版本的堆排序代码示例,使用了Python内置的列表操作:
```python
def heapify(arr, n, i):
largest = i
l = 2 * i + 1
r = 2 * i + 2
# 如果左孩子比根节点大
if l < n and arr[i] < arr[l]:
largest = l
# 如果右孩子比当前最大值大
if r < n and arr[largest] < arr[r]:
largest = r
# 如果最大值不是根节点,则交换并继续调整子树
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[i], arr = arr, arr[i] # 交换
heapify(arr, i, 0)
# 示例
arr = [12, 11, 13, 5, 6, 7]
heap_sort(arr)
print("排序后的数组:", arr)
```
阅读全文