aoc2023/day2/main.py

42 lines
711 B
Python
Raw Normal View History

2023-12-02 17:33:48 -05:00
fname = "input"
data = []
with open(fname) as f:
data = f.readlines()
"""
if the bag had been loaded with only
12 red cubes,
13 green cubes, and
14 blue cubes
"""
MAXIMUMS = {
"blue": 14,
"green": 13,
"red": 12
}
total = 0
for d in data:
possible = True
ginfo = d.split(":")
gid = int(ginfo[0].split("Game")[-1])
sets = ginfo[-1].split(";")
for s in sets:
d = s.split(",")
for item in d:
item = item.strip()
vals = item.split(" ")
color = vals[-1]
count = int(vals[0])
if count > MAXIMUMS[color]:
possible = False
if possible:
total = total + gid
print(total)