generated from public/aoc_template
67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
from collections import defaultdict
|
|
|
|
from tools.aoc import AOCDay
|
|
from tools.math import mul
|
|
from typing import Any
|
|
|
|
|
|
class Day(AOCDay):
|
|
inputs = [
|
|
[
|
|
(8, "input2_test"),
|
|
(2551, "input2"),
|
|
],
|
|
[
|
|
(2286, "input2_test"),
|
|
(62811, "input2"),
|
|
],
|
|
]
|
|
|
|
def part1(self) -> Any:
|
|
MAX = {
|
|
"red": 12,
|
|
"green": 13,
|
|
"blue": 14,
|
|
}
|
|
|
|
possible_sum = 0
|
|
|
|
for game in self.getInput():
|
|
game_name, draws = game.split(": ")
|
|
_, game_id = game_name.split()
|
|
|
|
impossible = False
|
|
for draw in draws.split("; "):
|
|
if impossible:
|
|
break
|
|
|
|
for color_draw in draw.split(", "):
|
|
count, color = color_draw.split()
|
|
if int(count) > MAX[color]:
|
|
impossible = True
|
|
|
|
if not impossible:
|
|
possible_sum += int(game_id)
|
|
|
|
return possible_sum
|
|
|
|
def part2(self) -> Any:
|
|
set_power_sum = 0
|
|
for game in self.getInput():
|
|
_, draws = game.split(": ")
|
|
|
|
min_color = defaultdict(int)
|
|
for draw in draws.split("; "):
|
|
for color_draw in draw.split(", "):
|
|
count, color = color_draw.split()
|
|
min_color[color] = max(min_color[color], int(count))
|
|
|
|
set_power_sum += mul(min_color.values())
|
|
|
|
return set_power_sum
|
|
|
|
|
|
if __name__ == "__main__":
|
|
day = Day(2023, 2)
|
|
day.run(verbose=True)
|