散列的键,其值总和为特定值

问题描述 投票:0回答:2

我有一个哈希:

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

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

Q2 Q5

请帮我解决这个问题。

ruby hash
2个回答
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 

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}
© www.soinside.com 2019 - 2024. All rights reserved.