ruby – 数组的所有可能的分布
发布时间：2020-12-16 19:22:25 所属栏目：百科 来源：网络整理
导读：我不太清楚如何表达这一点,所以我只举例. 如果我写： 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
|
我不太清楚如何表达这一点,所以我只举例.
如果我写： 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的提示更新了答案 （编辑：李大同） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |