ruby – 数组的所有可能的分布

我不太清楚如何表达这一点,所以我只举例.

如果我写:

some_method(["a","b"],3)

我希望它能恢复某种形式

[{"a" => 0,"b" => 3},{"a" => 1,"b" => 2},{"a" => 2,"b" => 1},{"a" => 3,"b" => 0}]

如果我传入

some_method(%w(a b c),2)

预期的回报值应该是

[{"a" => 2,"b" => 0,"c" => 0},"b" => 1,"c" => 1},{"a" => 0,"b" => 2,"c" => 2}]

描述这很难,所以如果你回答这个问题,请提前感谢!

解决方法

这是一种方法
def some_method ary,num
 permutations = (0..num).to_a.repeated_permutation(ary.size).select do |ary| 
    ary.reduce(:+) == num 
 end

 return permutations.map { |a| (ary.zip a).to_h }
end

p some_method ["a",3
#=> [{"a"=>0,"b"=>3},{"a"=>1,"b"=>2},{"a"=>2,"b"=>1},{"a"=>3,"b"=>0}]
p some_method %w(a b c),2
#=> [{"a"=>0,"b"=>0,"c"=>2},{"a"=>0,"b"=>1,"c"=>1},"b"=>2,"c"=>0},"c"=>0}]

根据@seph的提示更新了答案

相关文章

validates:conclusion,:presence=>true,:inclusion=>{...
一、redis集群搭建redis3.0以前,提供了Sentinel工具来监控各...
分享一下我老师大神的人工智能教程。零基础!通俗易懂!风趣...
上一篇博文 ruby传参之引用类型 里边定义了一个方法名 mo...
一编程与编程语言 什么是编程语言? 能够被计算机所识别的表...
Ruby类和对象Ruby是一种完美的面向对象编程语言。面向对象编...