剑指offer 63. 数据流中的中位数

题目描述

如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用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上进行排序,无须赋值保存!

猜你喜欢

转载自blog.csdn.net/Dby_freedom/article/details/84534888