ruby - 我如何使用 ruby​​ 生成数字组合？

Question

我需要使用 ruby​​ 生成数字组合。 例如:

arr = [1,2,3,4,5]

约束条件是，组合数字应包含数字5，长度至少为3或以上。 (即 125、521、1245 等。)。以上数组元素(取值1~5)在组合数中可能出现一次或两次或多次。

Answer 1

试试这个:

arr = [1, 2, 3, 4, 5]
arr = arr * 5
out = []
3.upto(5) do |i|
  arr.combination(i) do |c|
    out << c if c.include? 5
  end
end
out = out.uniq.sort
puts out.inspect

# yields 2531 elements:
# [[1, 1, 1, 1, 5], [1, 1, 1, 2, 5], ... [2, 3, 5], ... [5, 5, 5, 5, 5]]