堆排序 - topk问题（python语言实现）

堆排序 - topk问题（python语言实现）

现在有n个数，设计算法得到前k⼤的数。（k<n）
解决思路：
1.取列表前k个元素建⽴⼀个⼩根堆。堆顶就是⽬前第k⼤的数。
2. 依次向后遍历原列表，对于列表中的元素，如果⼩于堆顶，则忽略该元素；如果⼤于堆顶，则将堆顶更换为该元素，并且对堆进⾏⼀次调整；
3.遍历列表所有元素后，倒序弹出堆顶。

代码实现：

#创建一个比较排序函数，供下面函数调用
def sift(li, low, high):
    i = low
    j = 2 * i + 1
    tmp = li[low]
    while j <= high:
        if j + 1 <= high and li[j+1] < li[j]:
            j = j + 1
        if li[j] < tmp:
            li[i] = li[j]
            i = j
            j = 2 * i + 1
        else:
            break
        li[i] = tmp

#解决topk问题的主体函数
def topk(li, k):
    heap = li[0:k]
    for i in range((k-2)//2, -1, -1):
        sift(heap, i, k-1)
    # 1.建堆
    for i in range(k, len(li)-1):
        if li[i] > heap[0]:
            heap[0] = li[i]
            sift(heap, 0, k-1)

    # 2.遍历
    for i in range(k-1, -1, -1):
        heap[0], heap[i] = heap[i], heap[0]
        sift(heap, 0, i - 1)
    # 3.出数
    return heap