将splat参数从一个方法传递到另一个方法

时间:2014-02-01 02:41:45

标签: ruby arrays parameters splat

我想使用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参数从一个传递给另一个? (特别是对于某些特定范围......)

谢谢!

3 个答案:

答案 0 :(得分:5)

当您调用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

答案 1 :(得分:0)

或者,您可以将数字作为数组接受到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

答案 2 :(得分:0)

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