Add day 2

This commit is contained in:
Sijmen 2023-12-05 14:12:56 +01:00
parent 016a92f3cd
commit 2194015259
2 changed files with 37 additions and 0 deletions

2
.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
*.in
*.ex*

35
day02.py Normal file
View file

@ -0,0 +1,35 @@
import fileinput
from functools import reduce
def main():
part1 = 0
part2 = 0
for i, line in enumerate(fileinput.input()):
line = line.strip()
subsets = line.split(": ")[1].split("; ")
maxes = {}
part1_possible = True
for subset in subsets:
for balls in subset.split(", "):
count, color = balls.split()
count = int(count)
maxes[color] = max(maxes.get(color, 0), count)
part1_possible = (
part1_possible
and (color != "red" or count <= 12)
and (color != "green" or count <= 13)
and (color != "blue" or count <= 14)
)
if part1_possible:
part1 += i + 1
power = reduce(lambda x, y: x * y, maxes.values())
part2 += power
print(part1, part2)
main()