diff --git a/day15.py b/day15.py index c88cb40..1e3f2fb 100644 --- a/day15.py +++ b/day15.py @@ -10,16 +10,16 @@ def getGrid(lines: List[str], multiply: bool = False) -> Grid: for y, l in enumerate(lines): for x, v in enumerate(map(int, l)): g.set(Coordinate(x, y), v) - if multiply: - for x2 in range(5): - for y2 in range(5): - if x2 == 0 and y2 == 0: - continue - nv = v + x2 + y2 - if nv > 9: - nv -= 9 + if not multiply: + continue - g.set(Coordinate(size * x2 + x, size * y2 + y), nv) + for x2 in range(5): + for y2 in range(5): + if x2 == y2 == 0: + continue + nv = 1 + (v + x2 + y2 - 1) % 9 + + g.set(Coordinate(size * x2 + x, size * y2 + y), nv) return g