aoc

Advent of Code
git clone https://git.pngu.org/aoc
Log | Files | Refs | README

1_2023.py (572B)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
import sys

D = open(sys.argv[1]).read().strip()
p1 = 0
p2 = 0
for line in D.split("\n"):
    p1_digits = []
    p2_digits = []
    for i, c in enumerate(line):
        if c.isdigit():
            p1_digits.append(c)
            p2_digits.append(c)
        for d, val in enumerate(
            ["one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
        ):
            if line[i:].startswith(val):
                p2_digits.append(str(d + 1))
    p1 += int(p1_digits[0] + p1_digits[-1])
    p2 += int(p2_digits[0] + p2_digits[-1])
print(p1)
print(p2)