From 765df90cbcf8dd47f1c125319b24689beca1feb1 Mon Sep 17 00:00:00 2001 From: Alpha Chen Date: Mon, 24 Dec 2018 20:25:40 -0700 Subject: [PATCH] [2018][ruby][15.2] --- 2018/ruby/day_15.rb | 56 +++++++++++++++++++++++++++++---------------- 1 file changed, 36 insertions(+), 20 deletions(-) diff --git a/2018/ruby/day_15.rb b/2018/ruby/day_15.rb index a723bfb..99df432 100644 --- a/2018/ruby/day_15.rb +++ b/2018/ruby/day_15.rb @@ -35,7 +35,7 @@ class Combat unit = @map.fetch(turn.pos) target = @map.fetch(attack) target.hp -= @attack.fetch(unit.class) - @map.delete(attack) if target.hp < 0 + @map.delete(attack) if target.hp <= 0 end end @@ -423,19 +423,31 @@ class Goblin < Unit end def solve(input) - map = Map.parse(input) - combat = Combat.new(map, 3) + (3..Float::INFINITY).each do |elf_attack| + map = Map.parse(input) + outcome = simulate(map, elf_attack) + next if outcome.nil? + return outcome + end +end + +def simulate(map, elf_attack) + elf_count = map.units.count {|_, unit| unit.is_a?(Elf) } + combat = Combat.new(map, elf_attack) map, count = combat.fight.map.with_index .inject(nil) {|last,(map,count)| + return if map.units.count {|_, unit| unit.is_a?(Elf) } < elf_count # puts map, map.units.values.map(&:to_s).inspect [map, count] - } - outcome = map.units.values.map(&:hp).sum * count + } + return if map.units.count {|_, unit| unit.is_a?(Elf) } < elf_count + map.units.values.map(&:hp).sum * count end class TestSolve < Minitest::Test def test_solve - assert_equal 27730, solve(<<~INPUT) + # assert_equal 27730, solve(<<~INPUT) + assert_equal 4988, solve(<<~INPUT) ####### #.G...# #...EG# @@ -445,17 +457,18 @@ class TestSolve < Minitest::Test ####### INPUT - assert_equal 36334, solve(<<~INPUT) - ####### - #G..#E# - #E#E.E# - #G.##.# - #...#E# - #...E.# - ####### - INPUT - - assert_equal 39514, solve(<<~INPUT) + # assert_equal 36334, solve(<<~INPUT) + ######## + ##G..#E# + ##E#E.E# + ##G.##.# + ##...#E# + ##...E.# + ######## + #INPUT + + # assert_equal 39514, solve(<<~INPUT) + assert_equal 31284, solve(<<~INPUT) ####### #E..EG# #.#G.E# @@ -465,7 +478,8 @@ class TestSolve < Minitest::Test ####### INPUT - assert_equal 27755, solve(<<~INPUT) + # assert_equal 27755, solve(<<~INPUT) + assert_equal 3478, solve(<<~INPUT) ####### #E.G#.# #.#G..# @@ -475,7 +489,8 @@ class TestSolve < Minitest::Test ####### INPUT - assert_equal 28944, solve(<<~INPUT) + # assert_equal 28944, solve(<<~INPUT) + assert_equal 6474, solve(<<~INPUT) ####### #.E...# #.#..G# @@ -485,7 +500,8 @@ class TestSolve < Minitest::Test ####### INPUT - assert_equal 18740, solve(<<~INPUT) + # assert_equal 18740, solve(<<~INPUT) + assert_equal 1140, solve(<<~INPUT) ######### #G......# #.E.#...#