more cleaning
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
# 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)
|
||||
Reference in New Issue
Block a user