Solve Codyssi 14

Author: IsaacG

codyssi/problem14.py

@@ -0,0 +1,68 @@
+"""Codyssi Day N."""
+
+import queue
+import logging
+
+log = logging.info
+
+def solve(part: int, data: str) -> int:
+    """Solve the parts."""
+    rows = [[int(i) for i in line.split()]for line in data.splitlines()]
+    cols = [list(i) for i in zip(*rows)]
+    vals = {(x, y): val for y, row in enumerate(rows) for x, val in enumerate(row)}
+    steps = [(0, 1), (1, 0)]
+    if part == 1:
+        return min(
+            min(sum(i) for i in rows),
+            min(sum(i) for i in cols),
+        )
+    if part == 2:
+        target = (14, 14)
+    else:
+        target = (len(rows) - 1, len(cols) - 1)
+
+    safest = {}
+    todo = queue.PriorityQueue()
+    todo.put((0, (0, 0)))
+    while not todo.empty():
+        danger, pos = todo.get()
+        danger += vals[pos]
+        if pos == target:
+            return danger
+        if pos in safest and safest[pos] <= danger:
+            continue
+        safest[pos] = danger
+        for step in steps:
+            next_pos = tuple(i + j for i, j in zip(pos, step))
+            if next_pos in vals:
+                todo.put((danger, next_pos))
+    raise RuntimeError("No solution found.")
+
+
+TEST_DATA = """\
+3 3 1 7 8 4 1 3 1 7 7 6 7 8 7 8 2 7 7 1
+9 9 7 6 3 6 9 4 9 2 6 4 5 7 3 9 3 7 5 6
+8 9 7 6 7 7 3 2 2 7 8 9 7 1 5 3 1 2 4 4
+9 2 8 2 3 5 9 2 6 5 7 8 1 6 7 3 6 7 9 6
+4 1 7 5 2 2 7 6 8 7 2 3 9 2 2 1 6 2 7 5
+2 9 1 2 9 9 1 2 2 9 3 7 4 5 3 3 7 1 9 4
+9 9 5 2 6 6 2 3 1 8 3 3 3 6 7 9 8 3 1 5
+8 4 8 7 2 1 7 9 8 7 3 7 9 1 8 5 2 5 2 8
+6 8 9 6 6 4 2 2 7 7 7 8 1 2 6 2 6 1 6 7
+3 8 8 6 9 9 2 7 8 5 4 1 8 8 5 8 3 5 6 6
+2 8 7 2 6 8 4 7 1 7 6 8 9 4 3 1 2 8 9 8
+6 2 9 7 7 2 8 7 9 5 6 6 8 2 8 4 4 8 2 2
+3 1 2 8 8 4 6 8 9 1 4 3 9 1 4 2 2 1 5 4
+5 2 6 7 2 7 3 9 2 1 7 6 1 2 4 2 1 1 5 9
+3 6 8 9 4 4 7 7 3 3 4 8 6 1 2 9 7 2 9 6
+9 4 5 5 7 4 1 7 7 1 3 2 3 8 1 7 6 3 1 9
+5 3 8 3 1 1 5 3 1 5 9 2 3 6 6 4 4 8 5 3
+6 3 8 2 9 7 3 6 4 3 2 8 6 9 8 1 2 7 1 5
+4 1 2 4 8 7 7 1 8 7 4 4 5 7 2 3 3 8 3 3
+1 5 7 3 3 5 1 5 4 1 1 1 9 2 1 4 6 5 6 3"""
+
+TESTS = [
+    (1, TEST_DATA, 73),
+    (2, TEST_DATA, 94),
+    (3, TEST_DATA, 120),
+]

codyssi/solutions.txt

@@ -37,3 +37,6 @@
 13.1 8543
 13.2 6130
 13.3 7726
+14.1 206
+14.2 98
+14.3 315