题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
思路:
这道题没什么难度,省点事,直接调用python list的.sort()
方法,依据len(num)
是奇数还是偶数来输出中位数;
参考答案:
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.input_list = []
def Insert(self, num):
# write code here
self.input_list.append(num)
self.input_list.sort()
def GetMedian(self, M):
# write code here
length = len(self.input_list)
if length % 2 == 0:
mid = (self.input_list[length//2] + self.input_list[length//2-1])/2.0
else:
mid = self.input_list[length//2]
return mid
Note
- 本道题自己开始一直误以为直接使用
sorted(self.input_list)
就完成了对self.input_list
的排序,实际上是错误的,sorted()方法不会对原list进行操作,排序结果需要赋值保存,只有list.sort()是直接在list上进行排序,无须赋值保存!