def adjust_heap(arr, i, size):
lchild = 2 * i + 1
rchild = 2 * i + 2
max = i
if i < size / 2:
if lchild < size and arr[lchild] > arr[max]:
max = lchild
if rchild < size and arr[rchild] > arr[max]:
max = rchild
if max != i:
arr[max], arr[i] = arr[i], arr[max]
adjust_heap(arr, max, size)
def build_heap(arr, size):
for i in range(0, round(size / 2))[::-1]:
adjust_heap(arr, i, size)
def heap_sort(arr):
size = len(arr)
build_heap(arr, size)
for i in range(0, size)[::-1]:
arr[0], arr[i] = arr[i], arr[0]
adjust_heap(arr, 0, i)
return arr
排序算法(七):堆排序
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
推荐阅读更多精彩内容
- 选择排序 对于任何输入,时间为O(n*n); 冒泡排序 最优(对于升序的数组,因为加入了一个跳出判断):O(n),...
- 一般来说,这三者的时间复杂度O(NlogN),空间复杂度O(n)优化后,空间复杂度均可为O(1),如下文中的快排和...