Zwróć nieparzyste elementy z / bez bloku - rubin, blok, plon

Próbuję zwrócić nieparzyste elementy za pomocą tej metody

def odd_elements(array)
retArr = Array.new
array.each_with_index do |item, index|
if block_given?
yield(item)
else
retArr << item
end if index % 2 != 0
end

return retArr
end

żebym mógł przejść oba te testy

it "should yield odd elements" do
res = odd_elements([1,2,3,4,5,6])
res.should be_an_instance_of Array
res.should have(3).items
res.should include(2)
res.should include(4)
res.should include(6)
end

it "should yield" do
res = odd_elements([1,2,3,4,5,6]) {|x| x**2 }
res.should be_an_instance_of Array
res.should have(3).items
res[0].should == 4
res[1].should == 16
res[2].should == 36
end

ale zawodzę w drugim. Wygląda na to, że nie rozumiem, jak się poddać i nie udało mi się tego zrobić w ciągu dwóch godzin, próbując tylu różnych rzeczy. Czy możesz mi wyjaśnić, dlaczego to nie działa?

Odpowiedzi:

yield zwraca wartość, do której chcesz dodać retArr:

def odd_elements(array)
retArr = []
array.each_with_index do |item, index|
retArr << (block_given? ? yield(item) : item) if index % 2 != 0
end
retArr
end

Osobiście wolałbym Enumerator (jeśli korzystasz z Ruby 1.9.3+), nieco wolniej, ale prawdopodobnie bardziej czytelny i elastyczny:

def odd_elements(array)
Enumerator.new do |e|
odd = false
array.each do |x|
e << x if odd
odd = ! odd
end
end
end

a = [1,2,3,4]

odd_elements(a).to_a #=> [2, 4]
odd_elements(a).map { |x| x * 2 } #=> [4, 8]

odd_elements([1,2,3,4,5,6]){|x|p x**2 } #note the p
#=> 4
#=> 16
#=> 36

Daje to dobrze, ale nigdy nie zapisujesz wyniku w tablicy.

odd_elements([1,2,3,4,5,6]).map {|x| x**2 }

zapisałby wynik w tablicy.