Files
AoC/2023/Day-02/day02-2.py
2023-12-02 10:24:45 +00:00

32 lines
842 B
Python
Executable File

#!/usr/bin/python3
from math import prod
def main():
total = 0
with open('input.txt', 'r') as input_file:
for line in input_file:
total_colours = {
'red': 0,
'green': 0,
'blue': 0,
}
line = line.strip()
game_res = [result.split(',') for result in line.split(':')[1].split(';')]
for result in game_res:
for colour in result:
num_colour = int(colour.strip().split(' ')[0])
colour_name = colour.strip().split(' ', 1)[1]
if num_colour > total_colours[colour_name]:
total_colours[colour_name] = num_colour
total += prod(total_colours.values())
print(total)
if __name__ == '__main__':
main()