Question

我有一个数组，其元素是不同大小的数组。例如：

[[3],[11,2],[11,2],[3]]

我想找到嵌套数组中所有单个项的排列。对于上面的数组，我想要一个返回值：

[
  [3, 11, 11, 3],
  [3, 11, 2, 3],
  [3, 2, 11, 3],
  [3, 2, 2, 3]
]

我有一个有效的解决方案，但似乎特别啰嗦：

array = [[3],[11,2],[11,2],[3]]
array.product(*array).map { |e| e.drop(1) }.uniq

我应该如何实现递归方法，以及它将如何工作？我无法绕过这个。

Answer 1

解决此问题的传统方法是使用方法Array#product和Array#drop。

arr = [[3], [11,2], [11,2,7], [4]]

arr.first.product(*arr.drop(1))
  #=> [[3, 11, 11, 4], [3, 11, 2, 4], [3, 11, 7, 4],
  #    [3, 2, 11, 4], [3, 2, 2, 4], [3, 2, 7, 4]]

如果arr的任何元素包含重复项，则返回值也将包含重复项。如果不需要重复项，请使用

arr.map(&:uniq).first.product(*arr.drop(1))

然而，提问者要求递归解决方案。这可以写成如下：

def prod(arr)
  return arr if arr.size == 1
  t = prod(arr.drop(1))
  arr.first.flat_map { |x| t.map { |a| [x] + a } }
end

prod arr
  #=> [[3, 11, 11, 4], [3, 11, 2, 4], [3, 11, 7, 4],
  #    [3, 2, 11, 4], [3, 2, 2, 4], [3, 2, 7, 4]]

如何在Ruby

1 个答案: