41 lines
912 B
Python
41 lines
912 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
digit_map = {
|
||
|
|
'zero': 0,
|
||
|
|
'one': 1,
|
||
|
|
'two': 2,
|
||
|
|
'three': 3,
|
||
|
|
'four': 4,
|
||
|
|
'five': 5,
|
||
|
|
'six': 6,
|
||
|
|
'seven': 7,
|
||
|
|
'eight': 8,
|
||
|
|
'nine': 9
|
||
|
|
}
|
||
|
|
|
||
|
|
def find_digit_name(line):
|
||
|
|
for digit_name in digit_map.keys():
|
||
|
|
if line.startswith(digit_name):
|
||
|
|
return digit_map[digit_name]
|
||
|
|
return None
|
||
|
|
|
||
|
|
def find_digit(line, check_range):
|
||
|
|
for i in check_range:
|
||
|
|
if line[i].isdigit():
|
||
|
|
return int(line[i])
|
||
|
|
if (digit := find_digit_name(line[i:])) is not None:
|
||
|
|
return digit
|
||
|
|
|
||
|
|
|
||
|
|
# read input file
|
||
|
|
with open('input', 'r') as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
lines_sum = 0
|
||
|
|
for line in lines:
|
||
|
|
first_digit = find_digit(line, range(len(line)))
|
||
|
|
last_digit = find_digit(line, range(len(line) - 1, -1, -1))
|
||
|
|
num = str(first_digit) + str(last_digit)
|
||
|
|
lines_sum += int(num)
|
||
|
|
|
||
|
|
print(lines_sum)
|