Day 1 completed

day1/common.py

@@ -0,0 +1,11 @@
+def parse(filename: str) -> list[list[int]]:
+    elves = []
+    elf = []
+    with open(filename, "r") as f:
+        while line := f.readline():
+            if line == "\n":
+                elves.append(elf)
+                elf = []
+            else:
+                elf.append(int(line[:-1]))
+    return elves

day1/part1.py

@@ -0,0 +1,14 @@
+from common import parse
+
+
+def solve(elves: list[list[int]]) -> int:
+    return (sorted(map(sum, elves), reverse=True))[0]
+
+
+def main():
+    elves = parse("input")
+    print(solve(elves))
+
+
+if __name__ == "__main__":
+    main()

day1/part2.py

@@ -0,0 +1,14 @@
+from common import parse
+
+
+def solve(elves: list[list[int]]) -> int:
+    return sum(list(sorted(map(sum, elves), reverse=True))[0:3])
+
+
+def main():
+    elves = parse("input")
+    print(solve(elves))
+
+
+if __name__ == "__main__":
+    main()