🏠

y2022/p22.py

+import re
+
+
+text = open(0).read()
+p1, p2 = text.split('\n\n')
+
+grid = {}
+for y, row in enumerate(p1.splitlines()):
+    for x, cell in enumerate(row):
+        if cell.strip():
+            grid[complex(x, y)] = cell
+
+
+def render(grid):
+    xmin, *_, xmax = sorted(int(p.real) for p in grid)
+    ymin, *_, ymax = sorted(int(p.imag) for p in grid)
+    for y in range(ymin, ymax + 1):
+        for x in range(xmin, xmax + 1):
+            print(grid.get(complex(x, y), ' '), end='')
+        print()
+
+
+pos = min(grid, key=lambda p: (p.real == 0, p.imag))
+grid[pos] = '@'
+d = 1
+for n, b in zip(map(int, re.sub(r'[A-Z]', ' ', p2).split()), re.sub(r'\d', ' ', p2).split()):
+    for _ in range(n):
+        if grid.get(pos + d) in set('.<v>^'):
+            pos += d
+        elif grid.get(pos + d) == '#':
+            continue
+        elif grid.get(pos + d) == None:
+            lol = pos
+            while grid.get(lol - d) != None:
+                lol -= d
+            if grid[lol] == '.':
+                pos = lol
+            else:
+                continue
+    d *= {'R': 1j, 'L': -1j}[b]
+
+
+render(grid)
+print(4 * int(pos.real + 1) + 1000 * int(pos.imag + 1) + {1: 0, -1j: 1, -1: 2, 1j: 3}[d])