Question

我正在尝试使用此方法返回奇数元素

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

这样我就可以通过这两项测试

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

但我在第二次失败了。我似乎不明白如何屈服，我没有设法在两个小时内尝试这么多不同的事情。你能解释一下为什么它不起作用吗？

Answer 1

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

它产生正常，但你永远不会将结果存储在一个数组中。

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

会将结果保存在数组中。

Answer 2

yield会返回您似乎要添加到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

我个人更喜欢Enumerator（如果你使用的是Ruby 1.9.3+），有点慢，但可以说更具可读性和灵活性：

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]

返回带/不带块的奇数元素

2 个答案: