-
Notifications
You must be signed in to change notification settings - Fork 0
/
day2-p1.py
41 lines (31 loc) · 906 Bytes
/
day2-p1.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
f = open("input.txt", "r")
s = f.readlines()
red = 12
green = 13
blue = 14
r = 0
games = []
for l in s:
i = l.find(":")
games.append(l[(i+2):].split("; "))
for game in games:
r += (games.index(game) + 1)
for attempt in game:
red_count = 0
blue_count = 0
green_count = 0
for i in attempt.split(", "):
if "red" in i:
red_ball = int(i[:i.find("red")-1])
red_count += red_ball
if "blue" in i:
blue_ball = int(i[:i.find("blue")-1])
blue_count += blue_ball
if "green" in i:
green_ball = int(i[:i.find("green")-1])
green_count += green_ball
if red_count > red or blue_count > blue or green_count > green:
r -= (games.index(game)+1)
break
print(r)
#12 red, 13 green, 14 blue