老卫带你学---剑指offer刷题系列(63.数据流中的中位数)

63.数据流中的中位数

问题：

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

解决：

思想：

对于输入流问题，首先建立一个l列表作为输入流存储单元，然后通过Insert函数将数据流输入到存储单元中，最后从l中选取中间数。

python代码：

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.l=[]
    def Insert(self, num):
        # write code here
        self.l.append(num)
        self.l.sort()
    def GetMedian(self,data):
        # write code here
        n=len(self.l)
        if(n%2==1):
            return self.l[n//2]
        if(n%2==0):
            loc1,loc2=n//2-1,n//2
            mid=(self.l[loc1]+self.l[loc2])/2.0
            return mid