70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
with open('day21-input', 'r') as file:
|
|
data = [l.strip('\n') for l in file]
|
|
boss_hp = int(data[0].split()[-1])
|
|
boss_damage = int(data[1].split()[-1])
|
|
boss_armor = int(data[2].split()[-1])
|
|
|
|
player_hp = 100
|
|
player_damage = 0
|
|
player_armor = 0
|
|
|
|
def attack_damage(damage, armor):
|
|
return max(damage - armor, 1)
|
|
|
|
from math import ceil
|
|
def time_to_kill(damage, armor, hp):
|
|
return ceil(hp/attack_damage(damage, armor))
|
|
|
|
s_shop_w = """
|
|
Weapons: Cost Damage Armor
|
|
Dagger 8 4 0
|
|
Shortsword 10 5 0
|
|
Warhammer 25 6 0
|
|
Longsword 40 7 0
|
|
Greataxe 74 8 0
|
|
""".split('\n')[2:-1]
|
|
|
|
s_shop_a = """
|
|
Armor: Cost Damage Armor
|
|
Leather 13 0 1
|
|
Chainmail 31 0 2
|
|
Splintmail 53 0 3
|
|
Bandedmail 75 0 4
|
|
Platemail 102 0 5
|
|
""".split('\n')[2:-1]
|
|
|
|
s_shop_r = """
|
|
Rings: Cost Damage Armor
|
|
Damage +1 25 1 0
|
|
Damage +2 50 2 0
|
|
Damage +3 100 3 0
|
|
Defense +1 20 0 1
|
|
Defense +2 40 0 2
|
|
Defense +3 80 0 3
|
|
""".split('\n')[2:-1]
|
|
|
|
weapons = [[int(i) for i in w.split()[-3:]] for w in s_shop_w]
|
|
armors = [[int(i) for i in a.split()[-3:]] for a in s_shop_a] + [[0,0,0]]
|
|
rings = [[int(i) for i in r.split()[-3:]] for r in s_shop_r] + [[0,0,0], [0,0,0]]
|
|
|
|
def strategy_works(items):
|
|
# cost = sum([i[0] for i in items])
|
|
damage = sum([i[1] for i in items])
|
|
armor = sum([i[2] for i in items])
|
|
return time_to_kill(damage, boss_armor, boss_hp) <= time_to_kill(boss_damage, armor, player_hp)
|
|
|
|
from itertools import combinations
|
|
good_strategies = []
|
|
bad_strategies = []
|
|
for w in weapons:
|
|
for a in armors:
|
|
for r1, r2 in combinations(rings, 2):
|
|
strat = (w, a, r1, r2)
|
|
if strategy_works(strat):
|
|
good_strategies.append(strat)
|
|
else:
|
|
bad_strategies.append(strat)
|
|
|
|
print(min([sum([i[0] for i in items]) for items in good_strategies])) # Part 1
|
|
print(max([sum([i[0] for i in items]) for items in bad_strategies])) # Part 2
|