1.9_heap_topK_topK问题

堆排序 - topk 问题 (n 特别大时,很快)
现在有 n 个数,设计算法得到前 k 大的数

解决思路:
    1. 取列表前 k 个元素建立一个小根堆,堆顶就是目前第 k 大的数
    2. 依次向后遍历原列表,对于列表中的元素,如果小于堆顶,则忽略该元素
        如果大于堆顶,则将堆顶更换为该元素,并对堆进行一次调整
    3. 遍历列表所有元素后,倒序弹出堆顶
# 比较排序 - 通过比较来决定两个值是否交换
import random


def sift(li, low, high):

    i = low
    j = 2 * i + 1
    temp = li[low]

    while j <= high:
        if j + 1 <= high and li[j + 1] < li[j]:
            j = j + 1

        if li[j] < temp:
            li[i] = li[j]
            i = j
            j = 2 * i + 1
        else:
            break
    li[i] = temp


def topk(li, k):

    # 提取前 k 个元素建堆
    heap = li[0:k]
    for i in range((k - 2) // 2, -1, -1):
        sift(heap, i, k - 1)

    # 遍历剩下的元素,如果比堆顶大,就加进去,并对堆进行一次调整
    for i in range(k, len(li) - 1):
        if li[i] > heap[0]:
            heap[0] = li[i]
            sift(heap, 0, k - 1)

    # 挨个出数
    for i in range(k - 1, -1, -1):
        heap[0], heap[i] = heap[i], heap[0]
        sift(heap, 0, i - 1)

    return heap

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值