[路飞] 52——leetcode -面试题 17.20. 连续中值

148 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

面试题 17.20. 连续中值

题目分析

随机产生数字并传递给一个方法。你能否完成这个方法,在每次产生新值时,寻找当前所有值的中间值(中位数)并保存。

中位数是有序列表中间的数。如果列表长度是偶数,中位数则是中间两个数的平均值。

例如,

[2,3,4] 的中位数是 3

[2,3] 的中位数是 (2 + 3) / 2 = 2.5

设计一个支持以下两种操作的数据结构:

  • void addNum(int num) - 从数据流中添加一个整数到数据结构中。
  • double findMedian() - 返回目前所有元素的中位数。

思路讲解

  • 如果集合的个数是偶数个就是两个中间数相加除以2;
  • 如果是奇数个,中位数就是最中间的数。

示例

addNum(1)
addNum(2)
findMedian() -> 1.5
addNum(3) 
findMedian() -> 2

代码

/**
 * initialize your data structure here.
 */
var MedianFinder = function() {
    // 前面集合是一个大颈堆 后面集合是一个小颈堆,每次加值的时候进行排序
    this.item = []
};

/** 
 *  从数据流中添加一个整数到数据结构中
 * @param {number} num
 * @return {void}
 */
MedianFinder.prototype.addNum = function(num) {
    this.item.push(num)
};

/**
 * 返回目前所有元素的中位数
 * @return {number}
 */
MedianFinder.prototype.findMedian = function() {
    (this.item).sort((a, b) => a - b) // 排序
    // 判断奇数还是偶数
    if (this.item.length % 2 === 0) {
        return ((this.item)[this.item.length / 2] + (this.item)[this.item.length / 2 - 1]) / 2
    } else {
        return (this.item)[(this.item.length - 1) / 2]
    }
};

/**
 * Your MedianFinder object will be instantiated and called as such:
 * var obj = new MedianFinder()
 * obj.addNum(num)
 * var param_2 = obj.findMedian()
 */