Ruby：如何在数组中逐个比较元素？

Question

这是关于stackoverflow的第一个问题:)

考虑一个由任意整数组成的随机数组

a = [5, 10, 2, 3, 56]

我想编写一个代码，同样比较每个元素：索引 i (el_i) 的元素与索引 i + 1 (el_i+1) 的元素（并继续与元素索引 i + 1 (el_i+1) 进行比较) 具有索引 i + 2 (el_i+2) 直到 a.length 的元素。然后我想根据比较的结果对元素求和。这是一个失败的尝试描绘它的例子：

def conditional_sum(array)
  sum = 0
  array.each_with_index do |element, i|
    if array[i] >= array[i + 1]
      sum += element
    else
      sum -= element
    end
  end
  sum
end

另一个是从前一个数组中创建 2 个数组并删除最后一个元素（我省去了获取它们的代码），然后比较同样具有相同索引的元素：

a = [5, 10, 2, 3, 56]
a_bis = [5, 10, 2, 3]
b = [10, 2, 3, 56]  

sum = 0
for i in [0..a_bis.length]
  if a_bis[i] >= b[i] 
    sum += a_bis[i] + b[i]
  else
    sum -= a_bis[i] + b[i]
  end
end
p sum

也不行...非常感谢您的帮助！

Answer 1

如果我理解了你的问题：

a = [5, 10, 2, 3, 56]

a.each_cons(2).sum do |first, second|
  first > second ? first - second : first + second
end
#=>87

#each_cons 会将它们组合成成对的连续元素：

a.each_cons(2) do |pair|
  p pair
end
#=>[5, 10]
#  [10, 2]
#  [2, 3]
#  [3, 56]

然后您可以对块内的每一对执行所需的计算。

Answer 2

您的第一个示例几乎可以正常工作。尝试它会引发错误：

`>=': comparison of Integer with nil failed (ArgumentError)

这是因为对于索引的最后一次迭代，您将最后一个值 56 与 nil 进行比较。即56 >= nil 对红宝石没有意义，因此错误。要修复它，您可以在没有最后一个元素的情况下进行迭代。与 ruby​​ 一样，它们有多种方法可以做到这一点。这是我的：

def conditional_sum(array)
  sum = 0
  # array[0...-1] will iterate without le last element
  array[0...-1].each_with_index do |element, i|
    if array[i] >= array[i + 1]
      sum += element
    else
      sum -= element
    end
  end
  sum
end

执行您想要的另一种解决方案是以下（更实用的样式）。使用#each_cons 选择每一对

def conditional_sum(array)
  array.each_cons(2)
    .sum {|i,j| i >= j ? i : -i}
end