Объединить массив массив во все возможные комбинации, только вперед, в Ruby - PullRequest
27 голосов
/ 08 марта 2011

У меня есть массив массивов, например:

[['1','2'],['a','b'],['x','y']]

Мне нужно объединить эти массивы в строку, содержащую все возможные комбинации всех трех наборов, только вперед. Я видел много примеров всех возможных комбинаций наборов в любом порядке, это не то, что я хочу. Например, я не хочу, чтобы какие-либо элементы в первом наборе следовали за вторым набором или какие-либо элементы в третьем наборе предшествовали первому или второму и т. Д. Таким образом, для приведенного выше примера, результат будет:

['1ax', '1ay', '1bx', '1by', '2ax', '2ay', '2bx', '2by']

Число массивов и длина каждого набора являются динамическими.

Кто-нибудь знает, как решить эту проблему в Ruby?

Ответы [ 3 ]

55 голосов
/ 08 марта 2011

Знай твой Array#product:

a = [['1','2'],['a','b'],['x','y']]
a.first.product(*a[1..-1]).map(&:join)
7 голосов
/ 08 марта 2011

Решено с использованием рекурсивного, так называемого «Динамического программирования» подход:

  • Для n-массивов объединить записи первого массива с каждым результатом в оставшихся (n-1) массивах
  • Для одного массива ответом является только этот массив

В коде:

def variations(a)
  first = a.first
  if a.length==1 then
    first
  else
    rest = variations(a[1..-1])
    first.map{ |x| rest.map{ |y| "#{x}#{y}" } }.flatten
  end
end

p variations([['1','2'],['a','b'],['x','y']])
#=> ["1ax", "1ay", "1bx", "1by", "2ax", "2ay", "2bx", "2by"]

puts variations([%w[a b],%w[M N],['-'],%w[x y z],%w[0 1 2]]).join(' ')
#=> aM-x0 aM-x1 aM-x2 aM-y0 aM-y1 aM-y2 aM-z0 aM-z1 aM-z2 aN-x0 aN-x1 aN-x2
#=> aN-y0 aN-y1 aN-y2 aN-z0 aN-z1 aN-z2 bM-x0 bM-x1 bM-x2 bM-y0 bM-y1 bM-y2
#=> bM-z0 bM-z1 bM-z2 bN-x0 bN-x1 bN-x2 bN-y0 bN-y1 bN-y2 bN-z0 bN-z1 bN-z2

Вы также можете изменить логику, и с осторожностью вы сможете реализовать это нерекурсивно. Но рекурсивный ответ довольно прост. :)

3 голосов
/ 21 марта 2015

Чисто, уменьшите с продуктом:

a = [['1','2'],['a','b'],['x','y']]
a.reduce() { |acc, n| acc.product(n).map(&:flatten) }.map(&:join)
#  => ["1ax", "1ay", "1bx", "1by", "2ax", "2ay", "2bx", "2by"]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...