From f1907db70c912c43c73e78827be822ef8b382a23 Mon Sep 17 00:00:00 2001
From: Andy Lu <luandy64@gmail.com>
Date: Fri, 1 Dec 2023 12:53:26 -0500
Subject: [PATCH] Add day 1 solution

---
 python/2023/README.org                | 11 ++++
 python/2023/src/andy_aoc_2023/day1.py | 84 +++++++++++++++++++++++++++
 2 files changed, 95 insertions(+)
 create mode 100644 python/2023/src/andy_aoc_2023/day1.py

diff --git a/python/2023/README.org b/python/2023/README.org
index 5d6dc09..792a7d0 100644
--- a/python/2023/README.org
+++ b/python/2023/README.org
@@ -1 +1,12 @@
 * 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
diff --git a/python/2023/src/andy_aoc_2023/day1.py b/python/2023/src/andy_aoc_2023/day1.py
new file mode 100644
index 0000000..a515526
--- /dev/null
+++ b/python/2023/src/andy_aoc_2023/day1.py
@@ -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)