剑指 Offer 41. 数据流中的中位数

题目链接:

https://leetcode-cn.com/problems/shu-ju-liu-zhong-de-zhong-wei-shu-lcof/

题意:

得到一个数据流中的中位数

题解:

解法1:双顶堆
经典解法:大顶堆存放小的一半元素,小顶堆存放大的一般元素。询问中位数时从两个堆的堆顶元素可以得出。在加入元素时,需要不断平衡两个堆的元素个数基本持平(差距不超过1)即可。
解法2:二分插入
思路与解法1一样,都是通过二分去降低查询的时间复杂度。
做法是在添加元素时,利用二分插入使得整体元素有序,中位数直接通过数组中间一个或两个元素得到。

代码:
class MedianFinder:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.arr=[]

    def addNum(self, num: int) -> None:
        if len(self.arr)==0:
            self.arr.append(num)
        else:
            l,r = 0,len(self.arr)-1
            while l<=r:
                m =int((l+r)/2)
                if self.arr[m]<num:
                    l=m+1
                else:
                    r=m-1
            r =max(r,0)
            while self.arr[r]<num:
                r+=1
                if r==len(self.arr):
                    break
            self.arr.insert(r,num)

    def findMedian(self) -> float:
        if len(self.arr)&1 :
            return self.arr[int(len(self.arr)/2)]
        else:
            return (self.arr[int(len(self.arr)/2)]+self.arr[int(len(self.arr)/2)-1])/2


# Your MedianFinder object will be instantiated and called as such:
# obj = MedianFinder()
# obj.addNum(num)
# param_2 = obj.findMedian()
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值