Advent-of-Code/AdventOfCode-2018
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add part 2 solution (check out the previous commit for part 1.)

Browse Source
This commit is contained in:
Danila Fedorin 2018-12-15 03:06:32 -08:00
parent b7518c96a1
commit 6ff1d97c91
1 changed files with 29 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
day15.cr
View File

@ -213,9 +213,10 @@ end
def step(map, units) def step(map, units)
sort_reading_order(units) sort_reading_order(units)
index = 0 index = 0
elf_death_count = 0
while index < units.size while index < units.size
return true if (units.count &.[:type].==(:elf)) == 0 return nil if (units.count &.[:type].==(:elf)) == 0
return true if (units.count &.[:type].==(:goblin)) == 0 return nil if (units.count &.[:type].==(:goblin)) == 0
unit = units[index] unit = units[index]
unit = moved_unit(map, units, unit) unit = moved_unit(map, units, unit)
@ -224,6 +225,7 @@ def step(map, units)
if attacked_u if attacked_u
attacked_index, attacked_unit = attacked_u attacked_index, attacked_unit = attacked_u
if attacked_unit[:hp] <= 0 if attacked_unit[:hp] <= 0
elf_death_count += 1 if attacked_unit[:type] == :elf
units.delete_at attacked_index units.delete_at attacked_index
index -= 1 if index >= attacked_index index -= 1 if index >= attacked_index
else else
@ -233,7 +235,7 @@ def step(map, units)
index += 1 index += 1
end end
return false return elf_death_count
end end
def check_win(units) def check_win(units)
@ -252,11 +254,21 @@ def winning_score(units, type)
units.select(&.[:type].==(type)).sum &.[:hp] units.select(&.[:type].==(type)).sum &.[:hp]
end end
simulating_units = units.dup power = 3
loop do
simulating_units = units.dup.map do |unit|
next unit unless unit[:type] == :elf
{ x: unit[:x], y: unit[:y], hp: unit[:hp], type: :elf, power: power }
end
steps = 0 steps = 0
died = false
# draw(map, simulating_units) # draw(map, simulating_units)
until won = check_win(simulating_units) until won = check_win(simulating_units) || died
steps += 1 unless step(map, simulating_units) result = step(map, simulating_units)
if result
steps += 1
died = result > 0
end
# draw(map, simulating_units) # draw(map, simulating_units)
sort_reading_order(simulating_units) sort_reading_order(simulating_units)
# simulating_units.each do |unit| # simulating_units.each do |unit|
@ -264,4 +276,7 @@ until won = check_win(simulating_units)
# end # end
# puts "-- (end of roun #{steps})" # puts "-- (end of roun #{steps})"
end end
puts "#{steps} * #{winning_score(simulating_units, won)} = #{steps * winning_score(simulating_units, won)}" puts "(power: #{power}) #{steps} * #{winning_score(simulating_units, won)} = #{steps * winning_score(simulating_units, won)}"
break if !died && won != :goblin
power += 1
end