python实现的堆排序

      所谓堆是指序列a1 a2 a3 … an满足对任意的ai有ai >= a2i和ai >= a2i+1或者对任意的ai有ai <= a2i和ai <= a2i+1，这样的序列被称之为堆。当我们把序列以完全二叉树的形式来表示，ai是父结点，a2i和a2i+1是子结点，树的非叶结点的值都不大于或者不小于它的子结点的值。也就是堆是一棵具有一定性质（父结点值不小于/不大于子结点的值）的完全二叉树。

       完全二叉树的数据结构实现很简单，只需要用数组来实现，若将ai存放在数组下标为i的位置，则子结点的数组下标是2i和2i+1。数组下标从0开始，如果我们把ai存放到数组下标为i-1的位置，那么ai的子结点的数组下标是2i+1和2i+2。

       堆有个特点，堆顶元素（a1）是序列当中最大/最小值元素，堆排序是什么呢？堆排序就是通过把一个待排序的数组堆化来进行排序，也就是通过一些比较和交换操作把数组变成一个堆，输出堆顶（数组下标0位置元素），然后把数组当中最后一个元素存放到堆顶位置（数组下标0），通过一些调整再次把数组变成一个堆，然后再输出堆顶，以此类推，直至数组中只剩下一个元素，输出它，排序结束。

        通过上面的描述，实现堆排序需要解决两个问题：（1）如何由一个初始序列建成一个堆? （2）如何在输出堆顶元素之后，把剩余序列重新调整成一个堆？

        先来看怎么解决问题（2），假设堆顶元素取序列中的最小值，根据堆的定义，堆的任一子堆还是堆，假设输出堆顶元素之后，把数组当中最后一个元素ak放到了堆顶位置（取数组最后一个元素不会破坏左右子堆的堆性），原堆的左右子堆的堆顶元素是分别ai和aj，那么ai、aj、ak三者之中最小值将成为新堆的堆顶，如果是ai，则ak和ai换位，如果是aj，则ak和aj换位，如果ak仍然是最小的，什么都不用做，调整结束。如果发生了换位，那么左子堆或者右子堆的堆顶元素发生了改变，可能导致左子堆或者右子堆的堆性被破坏了，那么需要重复上面的操作把被破坏了的子堆重新调整成一个堆。这样自顶向下进行调整，要么操作到达某一级的子堆堆顶不需要交换父子元素而调整结束，要么操作到达了一个叶子结点而调整结束。

       再来看如何解决问题（1）。一个初始序列，我们把它看成一棵完全二叉树，把一个初始序列建成一个堆，其实就是把这棵完全二叉树堆化。如果完全二叉树的左右子树都是堆，那么就可以按照问题（2）的解决办法来把这棵完全二叉树调整成一个堆了，这也就是说，要把一棵完全二叉树建成一个堆，需要先把它的左右子树建成堆（这里多提一句，完全二叉树的子树也是完全二叉树）。前面的是induction，还是一个basis：一个结点的树是完全二叉树，它也是堆，也就是完全二叉树的叶子结点本身就是堆。由此，我们应该由初始序列的完全二叉树的最后一个非叶结点开始自右向左自底向上去依次把以它们为根结点的子树调整成堆。

        下面上代码：

def _siftup(heap, pos):
    newitem = heap[pos]
    chldpos = pos * 2 + 1
    endpos = len(heap)
    while chldpos < endpos:
        rchldpos = chldpos + 1
        if rchldpos < endpos and  heap[rchldpos] < heap[chldpos]:
            chldpos = rchldpos
        if heap[chldpos] < newitem:
            heap[pos] = heap[chldpos]
            pos = chldpos
            chldpos = pos * 2 + 1
        else:
            break
    heap[pos] = newitem

def heapify(x):
    n = len(x)
    for i in reversed(range(n // 2)):
        _siftup(x, i)

def heapsort(x):
    result = []
    heapify(x)
    while True:
        if len(x) > 1:
            result.append(x[0])
            x[0] = x.pop()
            _siftup(x, 0)
        else:
            result.append(x[0])
            break
    return result

print(heapsort([2, 1, 2, 3, 5, 4]))