ruby - 将 splat 参数从一种方法传递到另一种方法

Question

我想使用 splat 参数创建名为“add”和“subtract”的方法，如下所示：

def add(*numbers)
  numbers.inject(0) { |sum, n| sum + n }
end

def subtract(*numbers)
  numbers[0] - add(numbers[1..-1])
end

但它没有用。我应该怎么做才能将 splat 参数从一个传递到另一个？（特别是对于某些特定范围...）

谢谢！

Answer 1

调用add函数时，*在数组对象之前添加以将数组解包为多个参数。

def add(*numbers)
  numbers.inject(0) { |sum, n| sum + n }
end

def subtract(*numbers)
  numbers[0] - add(*numbers[1..-1])
  #                ^
end

subtract(9, 0, 1, 2) # => 6
subtract(9, 1) # => 8

Answer 2

或者，您可以将数字作为数组接受到 add 函数中。

def add(numbers)
  numbers.inject(0) { |sum, n| sum + n }
end

def subtract(*numbers)
  numbers[0] - add(numbers[1..-1])
end

一种更紧凑的方法（顺便说一下）是

def subtract(*numbers)
  numbers.shift - numbers.inject(:+)
end

Answer 3

def add *numbers
  [*numbers].flatten.inject 0, :+
end

def subtract *numbers
  numbers.shift - add(numbers)
end

subtract(9, 0, 1, 2) # ⇒ 6
subtract(9, 1)       # ⇒ 8