题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
分析
中位数就是在有序数组中排在中间的数,借用快速排序的partition方法,直接找到排在中间位置的数值即可。
要区分数组长度为奇数偶数的情况,分别计算中位数值,输入流数据是一个一个输入的,所以需要一个数组一次次保存。
找中位数时间O(n),空间O(1)(我这里多用了一个temp,其实可以省略的)
class Solution:
def __init__(self):
self.arr=[]
def GetMedian(self, data):
# write code here
def partition(a, low, high):
pivot = a[low] #枢轴为第一个元素
while low < high:
#从右向左找比枢轴元素小的数,移到左边(枢轴整好空一个位置出来,所以放到左边)
while low < high and a[high] >= pivot:
high -= 1
a[low] = a[high]