数据流中的中位数
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
思路
不同的数据结构对应着不同的解法,平时做题的时候要时刻想着该选用什么数据结构
数据结构 | 插入的时间复杂度 | 得到中位数的时间复杂度 |
---|---|---|
没有排序的数组 | O(1) | O(n) |
排序的数组 | O(n) | O(1) |
排序链表 | O(n) | O(1) |
二叉搜索树 | 平均O(logn),最差O(n) | 平均O(logn),最差O(n) |
AVL树 | O(logn) | O(1) |
最大堆和最小堆 | O(logn) | O(1) |
代码
var arr = [];
function Insert(num)
{
// write code here
arr.push(num);
arr.sort();
return arr;
}
function GetMedian(){
// write code here
var mid = Math.floor(arr.length / 2);
if((arr.length & 1) === 0){
var n1 = arr[mid];
var n2 = arr[mid - 1];
return (n1 + n2) / 2;
} else {
return arr[mid]
}
}