Ruby method overloading

Question

So I have to implement two different situations. One is a method that multiplies two numbers, and can also multiply more than 2 numbers.

I'm using the following:

def multiply(arr)
    arr.reduce(1, :*)
end

So far it works out fine if I unit test using an array input. Is there anyway to do this so my method can take in just two values, or an array, and return the relevant results? Is there also a way to implement this without even using an array input?

Consider using splat operator: def multiply(*arr); ...; end — samuil
– samuil, Commented May 8, 2014 at 12:49

Uri Agassi · Accepted Answer · 2014-05-08 12:49:29Z

3

Use the splat operator:

def multiply(*arr)
  arr.reduce(1, :*)
end

multiply(2, 3, 4, 5)
# => 120

If you want to also want to support input as an array, you can use flatten on arr:

def multiply(*arr)
  arr.flatten.reduce(1, :*)
end

multiply([2, 3, 4, 5])
# => 120 
multiply(10, 3, 5)
# => 150 
multiply(10, 3)
# => 30

answered May 8, 2014 at 12:49

Uri Agassi

37.5k16 gold badges82 silver badges96 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Ruby method overloading

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related