数学08--数据流中的中位数

数学08--数据流中的中位数-jz63

• ​​题目概述​​
• ​​解析&参考答案​​
• ​​注意事项​​
• ​​说明​​

题目概述

• 算法说明
  如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。
• 测试用例
  先后插入1，2，3
  其中位数分别为1，2，3

解析&参考答案

• 解析
  方法1: 排序后，按照中位数的方式计算并返回即可；
• 参考答案

vim jz63.go
package main

import (
    "fmt"
    "sort"
)

var arr []int

func Insert(num int) {
    arr = append(arr, num)
}

func GetMedian() float64 {
    if len(arr) == 0 {
        return 0
    }
    if len(arr) == 1 {
        return float64(arr[0])
    }
    sort.Ints(arr)
    length := len(arr)
    if (length & 1) == 1 {
        return float64(arr[length/2])
    } else {
        return (float64(arr[length/2]) + float64(arr[(length/2)-1])) / 2.0
    }
}

func main() {
    Insert(1)
    fmt.Println(GetMedian())
    Insert(3)
    fmt.Println(GetMedian())
    Insert(5)
    fmt.Println(GetMedian())
}

注意事项

1. to add

说明

1. 当前使用 go1.15.8
2. 参考​​牛客网--剑指offer​​ 标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

• 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
• 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解