public/aoc2021
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move getGrid inside, it's rainy

Browse Source
This commit is contained in:
Stefan Harmuth 2022-01-14 12:44:09 +01:00
parent 8552731beb
commit 6d77259384
1 changed files with 21 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
day15.py
View File

@ -4,10 +4,22 @@ from tools.grid import Grid
from typing import Any, List from typing import Any, List
def getGrid(lines: List[str], multiply: bool = False) -> Grid: class Day(AOCDay):
size = len(lines) inputs = [
[
(40, "test_input15"),
(386, "input15")
],
[
(315, "test_input15"),
(2806, "input15")
]
]
def getGrid(self, multiply: bool = False) -> Grid:
size = len(self.getInput())
g = Grid() g = Grid()
for y, l in enumerate(lines): for y, l in enumerate(self.getInput()):
for x, v in enumerate(map(int, l)): for x, v in enumerate(map(int, l)):
g.set(Coordinate(x, y), v) g.set(Coordinate(x, y), v)
if not multiply: if not multiply:
@ -23,26 +35,13 @@ def getGrid(lines: List[str], multiply: bool = False) -> Grid:
return g return g
class Day(AOCDay):
inputs = [
[
(40, "test_input15"),
(386, "input15")
],
[
(315, "test_input15"),
(2806, "input15")
]
]
def part1(self) -> Any: def part1(self) -> Any:
grid = getGrid(self.getInput()) grid = self.getGrid()
path = grid.getPath(Coordinate(0, 0), Coordinate(grid.maxX, grid.maxY), includeDiagonal=False, weighted=True) path = grid.getPath(Coordinate(0, 0), Coordinate(grid.maxX, grid.maxY), includeDiagonal=False, weighted=True)
return sum(grid.get(c) for c in path[:-1]) return sum(grid.get(c) for c in path[:-1])
def part2(self) -> Any: def part2(self) -> Any:
grid = getGrid(self.getInput(), True) grid = self.getGrid(True)
path = grid.getPath(Coordinate(0, 0), Coordinate(grid.maxX, grid.maxY), includeDiagonal=False, weighted=True) path = grid.getPath(Coordinate(0, 0), Coordinate(grid.maxX, grid.maxY), includeDiagonal=False, weighted=True)
return sum(grid.get(c) for c in path[:-1]) return sum(grid.get(c) for c in path[:-1])