Передача параметра пейнтбольного от одного метода к другому методу

Question

Я хотел бы создать методы, называемые «добавить» и «вычесть» с помощью параметра пейнтбольного, как показано ниже:

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

В качестве альтернативы вы можете принимать числа в функцию добавления в виде массива.

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