值总和为特定值的哈希键

时间:2019-03-17 09:00:00

标签: ruby hash

我有一个哈希:

a = {"Q1"=>1, "Q2"=>2, "Q5"=>3, "Q8"=>3}

我想从中检索一组键,以使它们的值之和等于某个数字,例如5。在这种情况下,输出应为:

Q2 Q5

请帮助我获取此信息。

2 个答案:

答案 0 :(得分:1)

只是出于好奇:

hash = {"Q1"=>1, "Q2"=>2, "Q5"=>3, "Q8"=>3}
arr = hash.to_a

1.upto(hash.size).
  lazy.
  find do |i|
    res = arr.combination(i).find do |h|
      h.map(&:last).sum == 5
    end
    break res if res
  end.tap { |result| break result.to_h if result }  
#⇒ {"Q2" => 2, "Q5" => 3}

答案 1 :(得分:1)

def find_combo(h, tot)
  arr = h.to_a
  (1..arr.size).find do |n|
    enum = arr.combination(n).find { |e| e.map(&:last).sum == tot }
    return enum.map(&:first) unless enum.nil?
  end
 end

h = {"Q1"=>1, "Q2"=>2, "Q5"=>3, "Q8"=>3}

find_combo(h, 5)   #=> ["Q2", "Q5"] 
find_combo(h, 2)   #=> ["Q2"] 
find_combo(h, 6)   #=> ["Q5", "Q8"] 
find_combo(h, 4)   #=> ["Q1", "Q5"] 
find_combo(h, 8)   #=> ["Q2", "Q5", "Q8"] 
find_combo(h, 9)   #=> ["Q1", "Q2", "Q5", "Q8"] 
find_combo(h, 10)  #=> nil