33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
def main(filename, bag, 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
|
|
|
|
return result
|
|
|
|
if __name__ == '__main__':
|
|
#red / green / blue
|
|
print(main("input.txt", [12, 13, 14])) |