generated from public/aoc_template
90 lines
2.4 KiB
Python
90 lines
2.4 KiB
Python
from tools.aoc import AOCDay
|
|
from tools.coordinate import Coordinate
|
|
from tools.grid import Grid
|
|
from typing import Any
|
|
|
|
|
|
def get_reflection_value(grid: Grid, ignore: int = 0) -> int:
|
|
# search horizontal
|
|
for x in grid.rangeX():
|
|
nx = x + 1
|
|
col = grid.get_column(x)
|
|
if col == grid.get_column(nx):
|
|
match = True
|
|
for dx in range(1, grid.maxX - x):
|
|
if x - dx < 0:
|
|
break
|
|
if grid.get_column(x - dx) != grid.get_column(nx + dx):
|
|
match = False
|
|
break
|
|
if match and nx != ignore:
|
|
return nx
|
|
|
|
# search vertical
|
|
for y in grid.rangeY():
|
|
ny = y + 1
|
|
row = grid.get_row(y)
|
|
if row == grid.get_row(ny):
|
|
match = True
|
|
for dy in range(1, grid.maxY - y):
|
|
if y - dy < 0:
|
|
break
|
|
if grid.get_row(y - dy) != grid.get_row(ny + dy):
|
|
match = False
|
|
break
|
|
if match and 100 * ny != ignore:
|
|
return 100 * ny
|
|
|
|
return -1
|
|
|
|
|
|
class Day(AOCDay):
|
|
inputs = [
|
|
[
|
|
(405, "input13_test"),
|
|
(30158, "input13_dennis"),
|
|
(32035, "input13"),
|
|
],
|
|
[
|
|
(400, "input13_test"),
|
|
(36474, "input13_dennis"),
|
|
(24847, "input13"),
|
|
],
|
|
]
|
|
|
|
def parse_input(self) -> list[Grid]:
|
|
return [Grid.from_data(pattern) for pattern in self.getMultiLineInputAsArray()]
|
|
|
|
def part1(self) -> Any:
|
|
grids = self.parse_input()
|
|
return sum([get_reflection_value(grid) for grid in grids])
|
|
|
|
def part2(self) -> Any:
|
|
grids = self.parse_input()
|
|
ref_value = 0
|
|
for grid in grids:
|
|
orig_ref = get_reflection_value(grid)
|
|
found = False
|
|
for x in grid.rangeX():
|
|
for y in grid.rangeY():
|
|
c = Coordinate(x, y)
|
|
grid.set(c, "." if grid.get(c) == "#" else "#")
|
|
|
|
v = get_reflection_value(grid, orig_ref)
|
|
if v > 0:
|
|
ref_value += v
|
|
found = True
|
|
break
|
|
|
|
grid.set(c, "." if grid.get(c) == "#" else "#")
|
|
|
|
if found:
|
|
break
|
|
|
|
return ref_value
|
|
|
|
|
|
if __name__ == "__main__":
|
|
day = Day(2023, 13)
|
|
day.run(verbose=True)
|