Skip to content

Instantly share code, notes, and snippets.

@konk303
Last active August 29, 2015 14:23
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save konk303/c9ae5c1a8af1bbae601e to your computer and use it in GitHub Desktop.
Save konk303/c9ae5c1a8af1bbae601e to your computer and use it in GitHub Desktop.
#!/usr/bin/env ruby
target = gets.to_i
count = gets.to_i
values = {}
count.times do |i|
values[gets.to_i] = 1
end
answer = 0
values_a = values.keys.sort
values_a.each_with_index do |vi, i|
values_a[i.succ, count].each do |vj|
need = target - vi - vj
# sortしてるので、`3つめより2つめが大きいか同じ == これ以上先に進んでも満たせる可能性がない`
break if vj >= need
answer = answer.succ if values[need]
end
end
puts answer
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment