代码
import random
class HeapSorter:
def __init__(self, arr):
self.arr = arr
def heapify(self, size, i):
p = i
v = self.arr[i]
while 2 * p + 1 < size:
left = 2 * p + 1
right = 2 * p + 2
# 二者最大值下标
if right < size and self.arr[right] > self.arr[left]:
left = right
# 不能下沉
if self.arr[left] <= v:
break
self.arr[p] = self.arr[left]
p = left
self.arr[p] = v
def build_heap(self):
n = len(self.arr)
# 为什么要循环
# 为了让父节点一定大于子节点
# 这样在寻找下一个最大节点的时候不用遍历整个树
for i in range(n // 2 + 1, -1, -1):
self.heapify(n, i)
def sort(self):
# 保证arr[0]是最大值
self.build_heap()
n = len(self.arr)
for i in range(n - 1, 0, -1):
# 将最大值移动到数组末尾
self.arr[i], self.arr[0] = self.arr[0], self.arr[i]
# 数组大小减1, 寻找下一个最大值
self.heapify(i, 0)
def print(self):
# 因为是最大值在最后面, 所以是升序遍历
for v in self.arr:
print(v, end=' ')
arr = []
for i in range(30):
v = random.randint(0, 100)
arr.append(v)
sorter = HeapSorter(arr)
sorter.sort()
sorter.print()
输出
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
python 随手写的堆排序
原文地址:https://www.jb51.cc/wenti/3281955.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。