Python:数据流中的中位数

来源:互联网 发布:开机显示便签损坏 知乎 编辑:程序博客网 时间:2024/06/06 17:26


牛客网上的剑指 offer的在线编程:

题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。

# -*- coding:utf-8 -*-'''数据流中的中位数题目描述如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。'''class Solution:    def __init__(self):        self.ch = []    def Insert(self, num):        # write code here        self.ch.append(num)        self.ch.sort()    def GetMedian(self, ch):        # write code here        if not self.ch:            return None        if len(self.ch) % 2 == 0:            k = len(self.ch) / 2            return (float(self.ch[k]) + float(self.ch[k - 1])) / float(2)        else:            k = (len(self.ch) + 1) / 2            return float(self.ch[k - 1])