public/aoc2023
1
0
Fork 0
generated from public/aoc_template
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Day 13 - minor cleanups

Browse Source
This commit is contained in:
Stefan Harmuth 2023-12-13 15:02:03 +01:00
parent 62b041bc0c
commit 425b190332
1 changed files with 13 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
day13.py
View File

@ -7,31 +7,33 @@ from typing import Any
def get_reflection_value(grid: Grid, ignore: int = 0) -> int: def get_reflection_value(grid: Grid, ignore: int = 0) -> int:
# search horizontal # search horizontal
for x in grid.rangeX(): for x in grid.rangeX():
nx = x + 1
col = grid.get_column(x) col = grid.get_column(x)
if col == grid.get_column(x + 1): if col == grid.get_column(nx):
match = True match = True
for dx in range(1, grid.maxX - x): for dx in range(1, grid.maxX - x):
if x - dx < 0: if x - dx < 0:
break break
if grid.get_column(x - dx) != grid.get_column(x + dx + 1): if grid.get_column(x - dx) != grid.get_column(nx + dx):
match = False match = False
break break
if match and x + 1 != ignore: if match and nx != ignore:
return x + 1 return nx
# search vertical # search vertical
for y in grid.rangeY(): for y in grid.rangeY():
ny = y + 1
row = grid.get_row(y) row = grid.get_row(y)
if row == grid.get_row(y + 1): if row == grid.get_row(ny):
match = True match = True
for dy in range(1, grid.maxY - y): for dy in range(1, grid.maxY - y):
if y - dy < 0: if y - dy < 0:
break break
if grid.get_row(y - dy) != grid.get_row(y + dy + 1): if grid.get_row(y - dy) != grid.get_row(ny + dy):
match = False match = False
break break
if match and 100 * (y + 1) != ignore: if match and 100 * ny != ignore:
return 100 * (y + 1) return 100 * ny
return -1 return -1
@ -60,16 +62,13 @@ class Day(AOCDay):
def part2(self) -> Any: def part2(self) -> Any:
grids = self.parse_input() grids = self.parse_input()
ref_value = 0 ref_value = 0
for i, grid in enumerate(grids): for grid in grids:
orig_ref = get_reflection_value(grid) orig_ref = get_reflection_value(grid)
found = False found = False
for x in grid.rangeX(): for x in grid.rangeX():
for y in grid.rangeY(): for y in grid.rangeY():
c = Coordinate(x, y) c = Coordinate(x, y)
if grid.get(c) == "#": grid.set(c, "." if grid.get(c) == "#" else "#")
grid.set(c, ".")
else:
grid.set(c, "#")
v = get_reflection_value(grid, orig_ref) v = get_reflection_value(grid, orig_ref)
if v > 0: if v > 0:
@ -77,10 +76,7 @@ class Day(AOCDay):
found = True found = True
break break
if grid.get(c) == "#": grid.set(c, "." if grid.get(c) == "#" else "#")
grid.set(c, ".")
else:
grid.set(c, "#")
if found: if found:
break break