题解 | #数据流中的中位数#

数据流中的中位数

http://www.nowcoder.com/practice/9be0172896bd43948f8a32fb954e1be1

# -*- coding:utf-8 -*-
class Solution:
    
    def __init__(self):
        self.heap_min = []
        self.heap_max =  []
    
    def Insert(self, num):
        # write code here
        import heapq
        if len(self.heap_min) > len(self.heap_max):
            heapq.heappush(self.heap_max, num)
        else:
            heapq.heappush(self.heap_min, -num)
        if len(self.heap_min)>0 and len(self.heap_max)>0 and -self.heap_min[0] > self.heap_max[0]:
            retMin = -heapq.heappop(self.heap_min)
            retMax = heapq.heappop(self.heap_max)
            heapq.heappush(self.heap_min, -retMax)
            heapq.heappush(self.heap_max, retMin)
        
    def GetMedian(self):
        # write code here
        if len(self.heap_min) > len(self.heap_max):
            return -self.heap_min[0]
        else:
            return (-self.heap_min[0]+self.heap_max[0])/2.0
全部评论

相关推荐

2024-12-21 10:42
已编辑
江西软件职业技术大学 Java
新宿站不停:该提升学历就提升学历,菜了就多练。没事找牛马公司虐自己是吧? 谁没事说自己“经验少”,这不自己把自己塞剎鼻hr嘴里找🐴吗
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务