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

题目概述

• 算法说明
  如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用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 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解