Jake Pullen
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
# with open(r'advent_of_code\2023\03\input.txt', 'r') as file:
|
|
# input = file.readlines()
|
|
|
|
# print(input)
|
|
|
|
test_input = '''467..114..
|
|
...*......
|
|
..35..633.
|
|
......#...
|
|
617*......
|
|
.....+.58.
|
|
..592.....
|
|
......755.
|
|
...$.*....
|
|
.664.598..'''
|
|
|
|
## remove when not test
|
|
input = test_input
|
|
## remove when not test
|
|
|
|
def get_directions(x_offset, y_offset):
|
|
for y in range(-1, 2):
|
|
for x in range(-1, 2):
|
|
yield x + x_offset, y + y_offset
|
|
|
|
valid_map = [[0 for _ in range(len(input[0]))] for _ in range(len(input))]
|
|
|
|
for i, row in enumerate(input[1:-1]):
|
|
for j, char in enumerate(row[1:-1]):
|
|
if not (char == "." or char.isnumeric()):
|
|
for x, y in get_directions(j + 1, i + 1):
|
|
valid_map[y][x] = 1
|
|
|
|
current_number = '0'
|
|
num_valid = False
|
|
sm = 0
|
|
|
|
for i, row in enumerate(input):
|
|
for j, char in enumerate(row):
|
|
if char.isnumeric():
|
|
num_valid = num_valid or valid_map[i][j]
|
|
current_number += char
|
|
else:
|
|
if num_valid:
|
|
sm += int(current_number)
|
|
num_valid = False
|
|
current_number = '0'
|
|
|
|
print(sm) |