Samuel Enocsson
549f0c4382
🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
import numpy as np
|
|
|
|
|
|
class Beam:
|
|
def __init__(self, start_loc: tuple, current_loc: tuple):
|
|
self.start_loc = start_loc
|
|
self.current_loc = current_loc
|
|
self.direction = None
|
|
self.beam_length = 0
|
|
self.energized = set()
|
|
self.active = True
|
|
|
|
def move(self, matrix: np.ndarray):
|
|
if self.direction is None:
|
|
self.direction = matrix[self.current_loc]
|
|
next_loc = get_next(self.current_loc, self.direction, matrix.shape)
|
|
if not (0 <= next_loc[0] < matrix.shape[0] and 0 <= next_loc[1] < matrix.shape[1]):
|
|
self.active = False
|
|
return None
|
|
self.current_loc = next_loc
|
|
self.beam_length += 1
|
|
return next_loc
|
|
|
|
def get_next(prev_loc: tuple, direction: str, shape: tuple):
|
|
match direction:
|
|
case '-':
|
|
return (prev_loc[0], prev_loc[1] - 1)
|
|
case '/':
|
|
return (prev_loc[0], prev_loc[1] + 1)
|
|
case '\\':
|
|
return (prev_loc[0] - 1, prev_loc[1])
|
|
case '|':
|
|
return (prev_loc[0] + 1, prev_loc[1])
|
|
case '.':
|
|
return prev_loc
|
|
|
|
file_path = '16-input.txt'
|
|
with open(file_path, 'r') as file:
|
|
lines = file.readlines()
|
|
# Split each line into individual characters
|
|
lines = [list(line.strip()) for line in lines]
|
|
|
|
# Build a NumPy matrix of all the lines
|
|
matrix = np.array(lines)
|
|
shape = matrix.shape
|
|
traverse = True
|
|
current_loc = (0, 0)
|
|
beams = []
|
|
beams.append(Beam(current_loc, current_loc))
|
|
while(traverse):
|
|
for beam in beams:
|
|
if beam.active:
|
|
next = beam.move(matrix)
|
|
|
|
|
|
|
|
|