如何使用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]] 

Answer 2

[編輯]功能的方法（需要Ruby 1.9）：

xs = 3.upto(5).flat_map do |length| 
    [1, 2, 3, 4, 5].repeated_permutation(length).select do |permutation| 
    permutation.include?(5) 
    end 
end 
xs.size # 2531 

Answer 3

arr = [1,2,3,4,5] 
combos = []    
for i in 3..arr.length 
    combos.push(arr.repeated_combination(i).to_a) 
end 
combos.flatten(1).select{|c|c.include?(5)} 

在這裏我創建一個臨時變量容器連擊將存儲數組中有3個或更多數字的每個組合。我然後過濾陣列只包含組合包含5.