-
Notifications
You must be signed in to change notification settings - Fork 0
/
part_one.py
64 lines (50 loc) · 2.23 KB
/
part_one.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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import sys
from game import Game
def main(print_output: bool = True):
try:
with open("input.txt") as input_file:
lines = input_file.readlines()
games: list[Game] = []
for line in lines:
# We create a new game object with the game_id and the max cubes
game = Game(
id=int(line.split()[1].removesuffix(":")),
)
# Remove the "Game ID: " part
rest_of_line = line.split(": ", 1)[1].removesuffix("\n")
# Directly divide the rest of the line into a list of cubes divided by ; and then by ,
hints = [data.split(", ") for data in rest_of_line.split("; ")]
# We iterate over the hints and update the max cubes for each color
for hint in hints:
for cube in hint:
amount, color = cube.split(" ")
if color == "red":
game.update_max_red_cubes(int(amount))
elif color == "green":
game.update_max_green_cubes(int(amount))
elif color == "blue":
game.update_max_blue_cubes(int(amount))
# We add the game to the list of games
games.append(game)
valid_games = []
# We then see which game has less or equal to 12 red cubes, 13 green cubes, and 14 blue cubes
for game in games:
if (
game.get_max_red_cubes() < 13
and game.get_max_green_cubes() < 14
and game.get_max_blue_cubes() < 15
):
valid_games.append(game)
# We calculate the sum of the game ids of the valid games
game_ids_sum = sum(game.get_id() for game in valid_games)
if print_output:
print(f"Sum of all game IDs: {game_ids_sum}")
sys.exit(0)
except FileNotFoundError:
print("The file input.txt was not found.")
sys.exit(1)
except Exception as e:
print(f"An error occurred: {e}")
sys.exit(1)
if __name__ == "__main__":
main()