solutions

1 files changed, 41 insertions, 1 deletions

diff --git a/22/solution.py b/22/solution.py
index 2859416..6df8379 100644
--- a/22/solution.py
+++ b/22/solution.py
@@ -1,2 +1,42 @@
+from functools import cache
+from collections import namedtuple
+
+Opponent = namedtuple('Opponent', ['hp', 'damage'])
+
+def part1(opponent: Opponent) -> int:
+    @cache
+    def min_cost(php: int, pmana: int, ohp: int, sld_timer: int, psn_timer: int, rch_timer: int) -> int | float:
+        psld = 0
+        def effects():
+            nonlocal ohp, pmana, psld, psn_timer, rch_timer, sld_timer
+            if sld_timer > 0:
+                sld_timer -= 1
+                psld = 7
+            else:
+                psld = 0
+            if psn_timer > 0:
+                psn_timer -= 1
+                ohp -= 3
+            if rch_timer > 0:
+                rch_timer -= 1
+                pmana += 101
+
+        def opponent_turn(cost: int) -> int | float:
+            effects()
+            nonlocal php
+            if ohp <= 0:
+                return cost
+            php -= min(1, opponent.attack - psld)
+            if php <= 0:
+                return float('inf')
+            return cost + min_cost(php, pmana, ohp, sld_timer, psn_timer, rch_timer)
+
+        effects()
+        if ohp <= 0:
+            return 0
+
 if __name__ == '__main__':
-    with open('input') as f:
\ No newline at end of file
+    opponent = Opponent(51, 9)
+
+    print(part1(opponent))
+    #print(part2(opponent))