题目
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用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