Создание перестановок из многомерного массива в Ruby

У меня есть следующий многомерный массив в Ruby:

[[1,2], [3], [4,5,6]]

Мне нужно иметь следующий вывод:

[[1,3,4], [1,3,5], [1,3,6], [2,3,4], [2,3,5], [2,3,6]]

Я попытался создать рекурсивную функцию, но мне не повезло.

Есть ли какие-либо функции Ruby, которые помогут с этим? Или это единственный способ сделать это рекурсивно?

Спасибо

Ответы

Ответ 1

Yup, Array#product делает именно это (Декартово произведение):

a = [[1,2], [3], [4,5,6]]
head, *rest = a # head = [1,2], rest = [[3], [4,5,6]]
head.product(*rest)
#=> [[1, 3, 4], [1, 3, 5], [1, 3, 6], [2, 3, 4], [2, 3, 5], [2, 3, 6]]

Другой вариант:

a.inject(&:product).map(&:flatten)
#=> [[1, 3, 4], [1, 3, 5], [1, 3, 6], [2, 3, 4], [2, 3, 5], [2, 3, 6]]