49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
|
def part1(lines):
|
||
|
|
res = []
|
||
|
|
|
||
|
|
for line in lines:
|
||
|
|
digits = [c for c in line if c.isnumeric()]
|
||
|
|
res.append(int(digits[0] + digits[-1]))
|
||
|
|
|
||
|
|
print(f"Part 1: {sum(res)}")
|
||
|
|
|
||
|
|
|
||
|
|
spelled_digits = { "one": "o1e", "two": "t2o", "three": "t3e", "four": "f4r", "five": "f5e", "six": "s6x", "seven": "s7n", "eight": "e8t", "nine": "n9e" }
|
||
|
|
|
||
|
|
|
||
|
|
def substitute_digits(s, trans):
|
||
|
|
res = s
|
||
|
|
for word, digit in trans.items():
|
||
|
|
res = res.replace(word, digit)
|
||
|
|
return res
|
||
|
|
|
||
|
|
|
||
|
|
def part2(lines):
|
||
|
|
res = []
|
||
|
|
|
||
|
|
for line in lines:
|
||
|
|
line = line.rstrip()
|
||
|
|
res_line = []
|
||
|
|
|
||
|
|
nline = ""
|
||
|
|
for c in line:
|
||
|
|
nline += c
|
||
|
|
nline = substitute_digits(nline, spelled_digits)
|
||
|
|
|
||
|
|
digits = [c for c in nline if c.isnumeric()]
|
||
|
|
r = int(digits[0] + digits[-1])
|
||
|
|
print(f"{line} => {nline}, {r}")
|
||
|
|
res.append(r)
|
||
|
|
|
||
|
|
print(f"Part 2: {sum(res)}")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import sys
|
||
|
|
infile = sys.argv[1]
|
||
|
|
|
||
|
|
with open(infile) as f:
|
||
|
|
lines = f.readlines()
|
||
|
|
part1(lines)
|
||
|
|
part2(lines)
|