703-Kth Largest Element in a Stream

最新推荐文章于 2021-02-25 19:01:22 发布

weixin_30375427

最新推荐文章于 2021-02-25 19:01:22 发布

阅读量110

点赞数

原文链接：http://www.cnblogs.com/kingshine007/p/11366194.html

版权

题目：返回数据流中的第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大元素即为堆顶元素

转载于:https://www.cnblogs.com/kingshine007/p/11366194.html