首页 > 其他 > 详细

703-Kth Largest Element in a Stream

时间:2019-08-16 22:17:25      阅读:117      评论:0      收藏:0      [点我收藏+]

题目:返回数据流中的第K大元素

class KthLargest(object):
    def __init__(self,k):
        self.min_heap = []
        self.k = k
    def add(self,data):
        if len(self.min_heap)<self.k:
            heapq.heappush(self.min_heap,data)
        else:
            heapq.heappushpop(self.min_heap,data)
    def get_kth(self):
        return heapq.nsmallest(1,self.min_heap)[0]

注:

构建一个小顶堆,大小为k,每次往里压入数据,第k大元素即为堆顶元素

703-Kth Largest Element in a Stream

原文:https://www.cnblogs.com/kingshine007/p/11366194.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!