Jake Pullen
91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
from functools import reduce
|
|
import operator
|
|
|
|
with open(r'advent_of_code\2023\02\input.txt', 'r') as file:
|
|
input = file.read()
|
|
|
|
# print(input)
|
|
test_input = '''Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green
|
|
Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue
|
|
Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red
|
|
Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red
|
|
Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green'''
|
|
|
|
amount_of_cubes = '12 red, 13 green, 14 blue'
|
|
amount_of_cubes = amount_of_cubes.split(', ')
|
|
# create a dictionary to store the amount of cubes
|
|
amount_of_cubes_dict = {}
|
|
for cube in amount_of_cubes:
|
|
# split the string into the amount and the colour
|
|
# first 2 characters are the amount
|
|
cube = [int(cube[:2]), cube[3:]]
|
|
amount_of_cubes_dict[cube[1]] = cube[0]
|
|
print(f'control: {amount_of_cubes_dict}')
|
|
|
|
def check_round(round_dict):
|
|
# Check if the keys exist in the dictionary, if not assign 0
|
|
red = round_dict.get('red', 0)
|
|
green = round_dict.get('green', 0)
|
|
blue = round_dict.get('blue', 0)
|
|
|
|
# Now compare with the amount_of_cubes_dict
|
|
if red > amount_of_cubes_dict['red'] or green > amount_of_cubes_dict['green'] or blue > amount_of_cubes_dict['blue']:
|
|
print('round', round_counter, 'is invalid')
|
|
return False
|
|
else:
|
|
print('round', round_counter, 'is valid')
|
|
return True
|
|
|
|
def highest_count_per_colour(game):
|
|
colour_counts = {}
|
|
groups = game.split(';')
|
|
for group in groups:
|
|
pairs = group.split(',')
|
|
for pair in pairs:
|
|
count, color = pair.strip().split(' ')
|
|
count = int(count)
|
|
if color not in colour_counts or count > colour_counts[color]:
|
|
colour_counts[color] = count
|
|
print(f'colour counts: {colour_counts}')
|
|
product_of_counts = reduce(operator.mul, colour_counts.values(), 1)
|
|
print(f'product of colour counts: {product_of_counts}')
|
|
return product_of_counts
|
|
#return 0
|
|
|
|
# break out each game to check if any of the games are valid
|
|
#games = test_input.split('\n')
|
|
games = input.split('\n')
|
|
game_counter = 0
|
|
score = 0
|
|
sum_of_powers = 0
|
|
for game in games:
|
|
game_counter += 1
|
|
game = game.split(': ')[1]
|
|
print('game', game_counter)
|
|
print(game)
|
|
round_counter = 0
|
|
rounds = game.split('; ')
|
|
powers = highest_count_per_colour(game)
|
|
sum_of_powers += powers
|
|
for round in rounds:
|
|
round_counter += 1
|
|
round = round.split(', ')
|
|
round_dict = {}
|
|
for cube in round:
|
|
cube = [int(cube[:2]), cube[2:]]
|
|
cube[1] = cube[1].replace(' ', '')
|
|
round_dict[cube[1]] = cube[0]
|
|
#print(round_dict)
|
|
|
|
if check_round(round_dict) == False:
|
|
break
|
|
else:
|
|
continue
|
|
else:
|
|
score += game_counter
|
|
print('game', game_counter, 'is valid')
|
|
continue
|
|
else:
|
|
print('score:', score)
|
|
print('sum of powers:', sum_of_powers)
|