47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
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 int(game_id) in result:
|
||
|
|
result.remove(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(":")
|
||
|
|
sets = line.rstrip().split(";")
|
||
|
|
for s in sets:
|
||
|
|
cubes = s.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)
|
||
|
|
|