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

数据流中的中位数

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
全部评论

相关推荐

叶扰云倾:进度更新,现在阿里云面完3面了,感觉3面答得还行,基本都答上了,自己熟悉的地方也说的比较细致,但感觉面试官有点心不在焉不知道是不是不想要我了,求阿里收留,我直接秒到岗当阿里孝子,学校那边的房子都退租了,下学期都不回学校,全职猛猛实习半年。这种条件还不诱人吗难道 然后现在约到了字节的一面和淘天的复活赛,外加猿辅导。华为笔试完没动静。 美团那边之前投了个base广州的,把我流程卡麻了,应该是不怎么招人,我直接简历挂了,现在进了一个正常的后端流程,还在筛选,不知道还有没有hc。
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务