大家好,欢迎来到IT知识分享网。
堆排序相对来说复杂一点,因此单写一篇文章来介绍,对其他排序算法感兴趣的,请移步:面试:经典排序算法思路及源码(收藏不看系列),下面言归正传说一说堆排序。
数据结构 – 堆
介绍堆排序之前,先介绍数据结构 – 堆,堆是一个完全二叉树,并且满足堆的性质:子结点的值总小于(或者大于)其父节点的值。一般来说,堆使用数组来存储,并且根据某个元素在数组中的位置可以推断出其子结点和父节点的位置。
第n个元素(从0开始计数)的左结点是2*n+1,右结点是2*n+2;其父结点是floor((n – 1)/2)
堆排序的思路
根据堆的定义可知,根是堆的最大元素或者最小元素,首先将数组调整为大顶堆,那么第0个元素就是最大元素,将第0个和最后一个元素(暂且称为第n个元素)交换位置,然后再将0~n-1个元素调整为堆,将第0个元素和第n-1元素交换,依次循环下去~
听起来很复杂,但只要花点时间,读完这篇文章,一定可以明白堆排序,下面一点点拆解~
如何将数组调整为堆
思路:从最后一个结点的父节点开始处理,假设一共有n个元素,那么最后一个结点的父节点是第(n-1)/2个元素,找到其子节点中值最大的节点,然后根其自身比较,如果自身值比子节点小,那么交换位置。
循环上述过程,从第(n-1)/2 一直遍历到第0个元素就完成了堆的构建。
Python代码如下所示:
def make_heap(array):
last_p = (len(array)-1)/2
while(last_p>=0):
child = 2*last_p+1
if(child+1 < len(array)):
if(array[child] < array[child+1]):
child = child+1
if(child < len(array) and array[child] > array[last_p]):
tmp = array[last_p]
array[last_p] = array[child]
array[child] = tmp
last_p=last_p-1
print(array[0])
完整逻辑
构建堆是堆排序中重要环节之一,构建完堆之后应该将第0个元素和最后一个元素交换位置,然后将前n-1个元素构建堆,再将第0个和倒数第二个交换…
其中涉及到一些细节,只有自己亲自动手编码才能掌握了解,这里给出一份DEMO
完整代码如下:
# -*- coding: UTF-8 -*-
import math
def make_heap(array,n):
last_p = (n-1)/2
while(last_p>=0):
child = 2*last_p+1
if(child+1 < n):
if(array[child] < array[child+1]):
child = child+1
if(child < n and array[child] > array[last_p]):
tmp = array[last_p]
array[last_p] = array[child]
array[child] = tmp
last_p=last_p-1
def swap_i_j(array,i,j):
array[i],array[j] = array[j],array[i]
def heap_sort(array):
n = len(array)
while(n>0):
make_heap(array,n)
print(n,array)
swap_i_j(array,0,n-1)
print(n,array)
n = n-1
array = [16,7,8,20,17,3]
heap_sort(array)
print(array)
免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://yundeesoft.com/49516.html