题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
思路:
第一种方法是使用大小顶堆,还不太理解这么做的原理,。。。
第二种方法通过数组的形式,添加后并排序;通过数组的长度寻找中位数的位置
代码:
# -*- coding:utf-8 -*-
from heapq import *
class Solution:
def __init__(self):
self.heaps=[],[]
def Insert(self, num):
# write code here
small,large=self.heaps
heappush(small,-heappushpop(large,num))
if len(large)<len(small):
heappush(large,-heappop(small))
def GetMedian(self,n=None):
# write code here
small,large=self.heaps
if len(large)>len(small):
return float(large[0])
return (large[0]-small[0])/2.0
class Solution:
def __init__(self):
self.arr=[]
def Insert(self,num):
self.arr.append(num)
self.arr.sort()
def GetMedian(self,n=None):
length=len(self.arr)
if length%2==1:
return self.arr[length//2]
return (self.arr[length//2]+self.arr[length//2-1])/2.0