看一百遍美女,美女也不一定是你的。但你刷一百遍算法,知识就是你的了~~
谁能九层台,不用累土起!
题目地址
题目
随机产生数字并传递给一个方法。你能否完成这个方法,在每次产生新值时,寻找当前所有值的中间值(中位数)并保存。
中位数是有序列表中间的数。如果列表长度是偶数,中位数则是中间两个数的平均值。
例如,
[2,3,4] 的中位数是 3
[2,3] 的中位数是 (2 + 3) / 2 = 2.5
设计一个支持以下两种操作的数据结构:
- void addNum(int num) - 从数据流中添加一个整数到数据结构中。
- double findMedian() - 返回目前所有元素的中位数。
示例:
1 2 3 4 5
| addNum(1) addNum(2) findMedian() -> 1.5 addNum(3) findMedian() -> 2
|
解题思路
- 我们在往数组中添加值的时候需要按序添加
- 输出中位数时对数组长度为奇数或偶数分情况处理
解题代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
|
var MedianFinder = function() { this.arr = [] };
MedianFinder.prototype.addNum = function(num) { let i = this.arr.findIndex(v=>v>num) if(i==-1){ this.arr.push(num) }else if(i==0){ this.arr.unshift(num) } else{ this.arr.splice(i,0,num) } };
MedianFinder.prototype.findMedian = function() { if(this.arr.length%2){ return this.arr[(this.arr.length-1)/2] }else{ return (this.arr[(this.arr.length/2)-1]+this.arr[this.arr.length/2])/2 } };
|
如有任何问题或建议,欢迎留言讨论!