力扣算法练习BM48—数据流中的中位数

题目

       如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

数据范围：数据流中数个数满足 1≤n≤1000  ，大小满足 1≤val≤1000 

进阶： 空间复杂度 O(n)  ， 时间复杂度 O(nlogn) 

示例1

输入：[5,2,3,4,1,6,7,0,8]

返回值："5.00 3.50 3.00 3.50 3.00 3.50 4.00 3.50 4.00 "

说明：数据流里面不断吐出的是5,2,3...,则得到的平均数分别为5,(5+2)/2,3...

解题思路

1.插入函数：每传入一次数字，就加入到列表中

2.获得中位数函数：

1）若列表中元素为奇数个，直接返回最中间的数

2）若列表中元素为偶数个，则计算中间两个数的平均值

3）返回结果为浮点型

题解

# -*- coding:utf-8 -*-
class Solution:
    num_list=[]
    result=[]
    # 插入：每传入一次数字，就加入到列表中
    def Insert(self, num):
        self.num_list.append(num)
        self.num_list.sort()
    # 获得中位数：
    def GetMedian(self):
        l=len(self.num_list)
        # 1.若列表中元素为奇数个，直接返回最中间的数
        if l%2==1:
            return round(self.num_list[l//2],2)
        # 2.若列表中元素为偶数个，则计算中间两个数的平均值    
        else:
            result=self.num_list[l//2]+self.num_list[l//2-1]
            return  result/2.0