Add day 1 solution
This commit is contained in:
parent
5f569ccb77
commit
f1907db70c
2 changed files with 95 additions and 0 deletions
|
@ -1 +1,12 @@
|
||||||
* Advent of Code Python
|
* Advent of Code Python
|
||||||
|
** Day 1
|
||||||
|
|
||||||
|
#+begin_src shell
|
||||||
|
source ~/virtualenvs/aoc-2023/bin/activate
|
||||||
|
cd ~/git/advent-of-code/python/2023
|
||||||
|
python src/andy_aoc_2023/day1.py src/andy_aoc_2023/day1_big
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
*** Answers
|
||||||
|
54573
|
||||||
|
54591
|
||||||
|
|
84
python/2023/src/andy_aoc_2023/day1.py
Normal file
84
python/2023/src/andy_aoc_2023/day1.py
Normal file
|
@ -0,0 +1,84 @@
|
||||||
|
from functools import reduce
|
||||||
|
from operator import add
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
NUMBERS = {
|
||||||
|
"1": "1",
|
||||||
|
"2": "2",
|
||||||
|
"3": "3",
|
||||||
|
"4": "4",
|
||||||
|
"5": "5",
|
||||||
|
"6": "6",
|
||||||
|
"7": "7",
|
||||||
|
"8": "8",
|
||||||
|
"9": "9",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
WORDS = {
|
||||||
|
"one": "1",
|
||||||
|
"two": "2",
|
||||||
|
"three": "3",
|
||||||
|
"four": "4",
|
||||||
|
"five": "5",
|
||||||
|
"six": "6",
|
||||||
|
"seven": "7",
|
||||||
|
"eight": "8",
|
||||||
|
"nine": "9",
|
||||||
|
}
|
||||||
|
|
||||||
|
BOTH = {**WORDS, **NUMBERS}
|
||||||
|
|
||||||
|
|
||||||
|
def get_numbers(line, parse_words=False):
|
||||||
|
if parse_words:
|
||||||
|
return re.findall(rf"(?=({'|'.join(BOTH)}))", line)
|
||||||
|
return re.findall("|".join(NUMBERS.keys()), line)
|
||||||
|
|
||||||
|
|
||||||
|
def line_to_number(line, parse_words=False):
|
||||||
|
string_nums = get_numbers(line, parse_words)
|
||||||
|
|
||||||
|
if parse_words:
|
||||||
|
lookup = BOTH
|
||||||
|
else:
|
||||||
|
lookup = NUMBERS
|
||||||
|
|
||||||
|
first = lookup[string_nums[0]]
|
||||||
|
last = lookup[string_nums[-1]]
|
||||||
|
number = first + last
|
||||||
|
return int(number)
|
||||||
|
|
||||||
|
|
||||||
|
def part1(lines):
|
||||||
|
nums = map(line_to_number,
|
||||||
|
lines)
|
||||||
|
|
||||||
|
answer = reduce(add,
|
||||||
|
nums)
|
||||||
|
print(answer)
|
||||||
|
|
||||||
|
|
||||||
|
def part2(lines):
|
||||||
|
nums = map(lambda x: line_to_number(x, parse_words=True),
|
||||||
|
lines)
|
||||||
|
|
||||||
|
answer = reduce(add,
|
||||||
|
nums)
|
||||||
|
|
||||||
|
print(answer)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
input_file = sys.argv[1]
|
||||||
|
|
||||||
|
with open(input_file) as infile:
|
||||||
|
input_lines = [
|
||||||
|
line.strip()
|
||||||
|
for line in infile.readlines()
|
||||||
|
]
|
||||||
|
|
||||||
|
part1(input_lines)
|
||||||
|
part2(input_lines)
|
Loading…
Reference in a new issue