如何获取数组中的下一个和上一个元素，Ruby

Question

我有这个数组，需要用它的前一个和下一个相乘来替换它的元素。

我执行以下操作：

   array.each_with_index{|element, index|
      next_element = array[index+1]
      previous_element = array[index-1]
    }

   array.map! {|a|
      if a == array.first
        a = a * next_element
      elsif a == array.last
        a = a * previous_element
      else
        a = next_element * previous_element
      end
      }

我希望得到以下结果：

array = [4, 1, 6, 7, 9, 3, 0]       #given array
array = [4, 24, 7, 54, 21, 0, 0]    #array replaced

我收到以下错误：

undefined local variable or method `next_element' for Arrays:Class

有没有一种简单的方法来获取给定数组元素的上一个和下一个元素？

我用的array.map!方法对吗？

Answer 1

这会起作用：

array = [4, 1, 6, 7, 9, 3, 0]

[nil, *array, nil].each_cons(3).map { |l, m, r| (l || m) * (r || m) }
#=> [4, 24, 7, 54, 21, 0, 0]

数组被nil值包围，所以每个元素都有邻居。each_cons(3)然后将每个元素与其邻居一起产生map，将左 ( l) 与右 ( r) 邻居相乘，m如果邻居之一恰好是，则回退到中间元素 ( ) nil。