python实现基于最小堆的topk

最新推荐文章于 2024-08-08 15:40:34 发布

NeilGY

最新推荐文章于 2024-08-08 15:40:34 发布

阅读量854

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： topk 堆排序 python

本文链接：https://blog.youkuaiyun.com/NeilGY/article/details/103393153

python 专栏收录该内容

5 篇文章

订阅专栏

本文深入探讨了Top-K算法的实现，通过构建堆数据结构来高效筛选出数组中最大的K个元素。文章详细介绍了堆的构建过程及算法流程，为读者提供了理解和应用Top-K算法的全面指南。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class MyTopK:
    def topK(self,k: int, nums: list):
        heap = self.buidHeap(nums[0:k])
        for i in range(k,len(nums)):
            if nums[i] > heap[0]:
                heap[0] = nums[i]
                heap[0],heap[-1] = heap[-1], heap[0]
                heap = self.buidHeap(heap)
        return heap

    def buidHeap(self,heap:list):
        # 数组中的数据是按照二叉树 层次遍历的，自下而上,遍历所有根节点， 将最小值放入根节点
        heap_len = len(heap)
        for i in range(heap_len // 2,-1,-1):
            left, right = 2*i + 1,2*i + 2
            if left < heap_len and heap[left] < heap[i]:
                heap[left], heap[i] = heap[i], heap[left]
            if right < heap_len and heap[right] < heap[i]:
                heap[right], heap[i] = heap[i], heap[right]
        return heap

        pass
if __name__ == '__main__':
    s = MyTopK().topK(5,[2,3,6,1,8,23,56,78,12])
    print(s)