diff options
Diffstat (limited to '2023/day2/solve2.py')
-rw-r--r-- | 2023/day2/solve2.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/2023/day2/solve2.py b/2023/day2/solve2.py new file mode 100644 index 0000000..c79821a --- /dev/null +++ b/2023/day2/solve2.py @@ -0,0 +1,23 @@ +def main(): + max_seen_colors = [] + power_sum = 0 + + with open("input", "r") as f: + for line in f: + current_game = {"red": 0, "green": 0, "blue": 0} + line_relevant = line.strip()[line.find(":") + 2:] + for grab in line_relevant.split("; "): + for items_saw in grab.split(", "): + count_and_color = items_saw.split(" ") + current_game[count_and_color[1]] = max(current_game[count_and_color[1]], int(count_and_color[0])) + + max_seen_colors.append(current_game) + + for elem in max_seen_colors: + power_sum += elem["red"] * elem["green"] * elem["blue"] + + print(power_sum) + + +if __name__ == '__main__': + main() |