From 56c4e033221775ad34a50d8f6d5276a7e3d61b55 Mon Sep 17 00:00:00 2001 From: Stefan Harmuth Date: Wed, 15 Dec 2021 12:49:06 +0100 Subject: [PATCH] day15: minimal code beautifications --- day15.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) 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