import 2023 aoc

2023/day2/day2.py

@@ -0,0 +1,45 @@
+from math import prod
+
+
+def part1(lines):
+    result = set()
+    for index, line in enumerate(lines):
+        game_id = index + 1
+        _, line = line.split(": ")
+        result.add(int(game_id))
+        hands = line.rstrip().split("; ")
+        for hand in hands:
+            colors = {"red": 0, "green": 0, "blue": 0}
+            cubes = hand.split(", ")
+            for cube in cubes:
+                n, color = cube.split()
+                colors[color] = int(n)
+            if colors["red"] > 12 or colors["green"] > 13 or colors["blue"] > 14:
+                # impossible configuration, remove this game_id from the result (if present)
+                result.discard(int(game_id))
+    print(f"Part 1: {sum(result)}")
+
+
+def part2(lines):
+    result = []
+    for line in lines:
+        colors = {"red": 0, "green": 0, "blue": 0}
+        _, line = line.split(": ")
+        hands = line.rstrip().split("; ")
+        for hand in hands:
+            cubes = hand.split(", ")
+            for cube in cubes:
+                n, color = cube.split()
+                colors[color] = max(colors[color], int(n))
+        result.append(prod(colors.values()))
+    print(f"Part 2: {sum(result)}")
+            
+
+if __name__ == "__main__":
+    import sys
+    infile = sys.argv[1]
+    with open(infile) as f:
+        lines = f.readlines()
+        part1(lines)
+        part2(lines)
+