Find the median of an array

前端 未结 4 1621
伪装坚强ぢ
伪装坚强ぢ 2021-01-25 17:25

I wrote some code that returns the median of an unsorted odd numbered array, but it does not return the median of an even numbered array.

I know that in order to find

4条回答
  •  日久生厌
    2021-01-25 18:15

    def get_median(values)
      quotient, remainder = values.sort!.length.divmod(2)
      medians = [values[quotient], values[quotient-(remainder+1) % 2]].uniq
      medians.length > 1 ? medians.map(&:to_f).reduce(:+) / 2 : medians[0]
    end
    

    This should work, but let me know if it does not.

    edit forgot to add .to_f to medians values. Here is my testing in Pry.

    [1] pry(main)> def get_median(values)
    [1] pry(main)*   quotient, remainder = values.sort!.length.divmod(2)
    [1] pry(main)*   medians = [values[quotient], values[quotient-(remainder+1) % 2]].uniq
    [1] pry(main)*   medians.length > 1 ? medians.map { |v| v.to_f }.reduce(:+) / 2 : medians[0]
    [1] pry(main)* end
    => nil
    [2] pry(main)> values = [1,2,3]
    => [1, 2, 3]
    [3] pry(main)> get_median values
    => 2
    [4] pry(main)> values << 4
    => [1, 2, 3, 4]
    [5] pry(main)> get_median values
    => 2.5
    [6] pry(main)> get_median values << -1
    => 2
    [7] pry(main)> values
    => [-1, 1, 2, 3, 4]
    [8] pry(main)> values << 11
    => [-1, 1, 2, 3, 4, 11]
    [9] pry(main)> get_median values << -1
    => 2
    [10] pry(main)> values
    => [-1, -1, 1, 2, 3, 4, 11]
    [11] pry(main)> values.delete_if { |v| v == -1}
    => [1, 2, 3, 4, 11]
    [12] pry(main)> values << -1
    => [1, 2, 3, 4, 11, -1]
    [13] pry(main)> get_median values
    => 2.5
    [14] pry(main)>
    

提交回复
热议问题