如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
看别人用两个堆,一个最大堆,一个最小堆,说的挺在理,其实理解起来也不难,
下面呢提供一种简单粗暴的方法,但是如果细细探究其时间复杂度,其实是O(n),但排序用的是内建函数,真实运行速度比较快,话不多说,上代码吧
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.res = []
def Insert(self, num):
# write code here
self.res.append(num)
def GetMedian(self, x):
# write code here
self.res.sort()
length = len(self.res)
if length % 2 == 0:
return (self.res[length/2-1] + self.res[length/2])/2.0
else:
return self.res[length/2]