python实现排序算法NB三人组之堆排序，以及topK问题实现

堆排序

代码如下：

'''
堆排序：
时间复杂度：O(n*logn)
实现较大数据n的topk问题

'''
def sift(li, low, high):
    '''

    :param li: 列表
    :param low: 根节点
    :param high: 最后一个节点
    :return:
    '''
    i = low
    j = 2*i+1 #j是i的左孩子
    temp = li[low] #把堆顶存起来
    while j<=high:  #只要j位置有数
        if j+1<=high and li[j+1]>li[j]:#如果右孩子有并且比左孩子大
            j = j+1 #j指向右孩子
        if li[j]>temp:
            li[i] = li[j] #大的子节点取代父节点的位置
            i = j          #往下走一层
            j = 2*i+1
        else:
            li[i] = temp  #把temp放在某一层领导位置上
            break
    else:
        li[i] = temp     #把temp放在叶子节点上

def heap_sort(li):
    n = len(li)
    for i in range((n-2)//2, -1, -1):
        sift(li, i, n-1)  #建立堆（完全二叉树）

    for i in range(n-1, -1 , -1):#i指向最后一个元素
        li[0], li[i] = li[i], li[0] #将根节点取出，放在最后一个节点位置，即排序
        sift(li, 0, i-1) #对去除根节点的新列表进行sift


'''
TOPk问题：
当n比较大时，取前 k 个数
1.堆排序方法排序比较快： O(nlogk)
2.冒泡，选择排序：kn
3.快排：n*logn
'''
def topk(li, k):
    n = len(li)
    heap = li[0:k]
    for i in range((k-2)//2, -1, -1):
        sift(heap, i, k-1) #建立k个数的堆

    for i in range(k, n):#遍历其它n-k个数与heap[0]比较
        if li[i] < heap[0]:
            heap[0] = li[i]
            sift(heap, 0, k-1)

    for i in range(k-1, -1, -1):
        heap[i], heap[0] = heap[0], heap[i]
        sift(heap, 0, i-1)
    return heap

li=[1, 3, 2, 4, 6, 7, 8, 5]

print(topk(li, 5))
heap_sort(li)
print(li)

结果如下：