Ruby Method, который возвращает медианное целое число в массиве

Question

Я хочу написать метод, который находит медиану заданного массива целых чисел. Может кто-то объяснить, что не так/происходит в моем коде. Если массив имеет нечетное число целых чисел, он возвращает средний элемент из отсортированного массива. Если массив имеет четное число целых чисел, верните среднее из средних двух элементов из отсортированного массива. Ниже мой код, который продолжает застревать в бесконечном цикле.

def median_finder(array) 
    array.sort 
    element_count = array.length 

    if element_count % 2 != 0 
    while element_count != 1 do 
     array.shift 
     array.pop 
    end 

    return array 
    else element_count % 2 == 0 
    while element_count != 2 do 
     array.shift 
     array.pop 
    end 
    median = ((array[0] + array[1])/2) 
    return median 
    end 
end 

Answer 1

ваше время петли не изменяет переменную element_count, и поэтому условие цикла не изменится, зацикливания. element_count всегда остается значением размера исходного массива. Быстрое исправление приведено ниже:

... 
while element_count != 1 do 
    array.shift 
    array.pop 
    element_count = array.length 
end 
... 
while element_count != 2 do 
    array.shift 
    array.pop 
    element_count = array.length 
end 

Answer 2

Ваше использование array.sort неверно. Это должно быть array.sort!.

Вместо:

if element_count % 2 != 0 

Использование:

if element_count.odd? 

Вместо:

array.shift 
array.pop 

Рассмотрим:

array = array[1..-2] 

else не принимает никаких условий. Это неправильно:

else element_count % 2 == 0 

Один из них будет работать:

else 

Или:

else # element_count % 2 == 0 

Будьте очень осторожны с этим видом логики:

element_count != 1 

Что если array имеет два e и вы их удаляете? element_count будет 0, и условный тест все равно не будет выполнен, поэтому код продолжит цикл.

Вместо этого используйте тест как:

element_count >1 

Answer 3

Если вы уже знаете, количество элементов в массиве, было бы проще, используя математику (= делением длины на два), а зацикливание:

def median_finder(array) 
    array = array.sort 
    if array.length.odd? 
    array[array.length/2] 
    else 
    array[array.length/2 - 1, 2].inject(:+)/2.0 
    end 
end 

inject(:+) - рубиновый способ делать сумму. array[array.length/2 - 1, 2] принимает два элемента в середине массива, так что вы могли бы на самом деле заменить эту строку с чем-то вроде этого:

median1, median2 = array[array.length/2 - 1, 2] 
    (median1 + median2)/2.0