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

题目描述:

 题解:

思路:维护一个有序队列,在获取中位数的时候,若为奇数,直接返回相应下标对应的值,若为偶数,返回中间两个下标对应数字的平均数。

1._init_初始化函数:创建一个list array,number保存array中元素的数量,初始化为0.

2.addNum函数,将num添加到array对应的位置,使添加后的array仍然有序。分三种情况:array为空,num大于等于array中最大的数字,num对应的位置在array中间。

3.findMedian函数:根据number的奇偶返回相应的值。

class MedianFinder:

    def __init__(self):
        self.array = []
        self.number = 0

    def addNum(self, num: int) -> None:
        if self.number == 0:
            self.array.append(num)
        elif num >= self.array[-1]:
            self.array.append(num)
        else:
            idx = 0
            while num >= self.array[idx]:
                idx = idx + 1
            self.array.insert(idx, num)
        self.number = self.number + 1

    def findMedian(self) -> float:
        if self.number == 0:
            return []
        if self.number % 2 == 1:
            mid = self.array[int((self.number - 1) / 2)]
        else:
            idx1 = int(self.number / 2)
            idx2 = idx1 - 1
            mid = (self.array[idx1] + self.array[idx2]) / 2
        return mid

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值