Mikaël Capelle
ce315b8778
Co-authored-by: Mikael CAPELLE <mikael.capelle@thalesaleniaspace.com> Co-authored-by: Mikaël Capelle <capelle.mikael@gmail.com> Reviewed-on: #3
34 lines
763 B
Python
34 lines
763 B
Python
from collections import defaultdict
|
|
from typing import Any, Iterator
|
|
|
|
from ..base import BaseSolver
|
|
|
|
|
|
def process(directions: str) -> dict[tuple[int, int], int]:
|
|
counts: dict[tuple[int, int], int] = defaultdict(lambda: 0)
|
|
counts[0, 0] = 1
|
|
x, y = (0, 0)
|
|
|
|
for c in directions:
|
|
match c:
|
|
case ">":
|
|
x += 1
|
|
case "<":
|
|
x -= 1
|
|
case "^":
|
|
y -= 1
|
|
case "v":
|
|
y += 1
|
|
case _:
|
|
raise ValueError()
|
|
|
|
counts[x, y] += 1
|
|
|
|
return counts
|
|
|
|
|
|
class Solver(BaseSolver):
|
|
def solve(self, input: str) -> Iterator[Any]:
|
|
yield len(process(input))
|
|
yield len(process(input[::2]) | process(input[1::2]))
|