Alter 2 files

Update main.py
Delete sample.txt
This commit is contained in:
akp 2024-12-06 16:38:02 +00:00
parent 7b894ac8c9
commit 617c90bb14
No known key found for this signature in database
GPG key ID: CF8D58F3DEB20755
2 changed files with 19 additions and 22 deletions

View file

@ -13,28 +13,35 @@ def find_start(g: grid.Grid) -> coord.Coordinate:
if g[pos] == "^":
return pos
assert False, "no start point found"
def modplus(x: int) -> int:
return (x + 1) % 4
dirs = [coord.Direction.Up, coord.Direction.Right, coord.Direction.Down, coord.Direction.Left]
dirs = [
coord.Direction.Up,
coord.Direction.Right,
coord.Direction.Down,
coord.Direction.Left,
]
class LoopEncounteredException(Exception):
pass
def trace(g: grid.Grid, guard_pos: coord.Coordinate, guard_direction: int) -> set[tuple[coord.Coordinate, int]]:
def trace(
g: grid.Grid, guard_pos: coord.Coordinate, guard_direction: int
) -> set[tuple[coord.Coordinate, int]]:
visited_sequence = set()
while guard_pos in g:
if (guard_pos, guard_direction) in visited_sequence:
raise LoopEncounteredException
visited_sequence.add((guard_pos, guard_direction))
nc = coord.add(guard_pos, dirs[guard_direction % 4].delta())
if nc in g and g[nc] == "#":
guard_direction = modplus(guard_direction)
@ -51,11 +58,11 @@ def one(instr: str) -> int:
def two(instr: str) -> int:
g = parse(instr)
start_pos = find_start(g)
seq = trace(g, start_pos, 0)
known_blocks = set()
for (pos, _) in tqdm(seq, file=sys.stderr):
assert pos in g, "pos off the rails"
g[pos] = "#"
@ -64,7 +71,7 @@ def two(instr: str) -> int:
except LoopEncounteredException:
known_blocks.add(pos)
g[pos] = "."
return len(known_blocks)
@ -81,4 +88,4 @@ if __name__ == "__main__":
if sys.argv[1] == "1":
print(one(inp))
else:
print(two(inp))
print(two(inp))

View file

@ -1,10 +0,0 @@
....#.....
.........#
..........
..#.......
.......#..
..........
.#..^.....
........#.
#.........
......#...