aoc2023/d02/run1.py

35 lines
920 B
Python

filename = "exemple1.txt"
#red / green / blue
bag = [12, 13, 14]
result = 0
with open(filename, 'r') as f:
for line in f.readlines():
game = line.split(": ")
p = game[1].replace("\n","").split("; ")
i = int(game[0].split(" ")[1])
result += i
#print(p)
# par pack
for c in p:
nb = [0, 0, 0]
#print(c)
# par jeton
for j in c.split(", "):
k = j.split(" ")
#print(j)
if k[1] == 'red':
nb[0] += int(k[0])
elif k[1] == 'green':
nb[1] += int(k[0])
elif k[1] == 'blue':
nb[2] += int(k[0])
else:
print("error")
if nb[0] > bag[0] or nb[1] > bag[1] or nb[2] > bag[2]:
result -= i
break
print(result)