advent24-llm / day11 /solution_gpt-4o.py
jerpint's picture
Add solution files
a4da721
raw
history blame
1.05 kB
def transform_stones(stones):
new_stones = []
for stone in stones:
if stone == 0:
new_stones.append(1)
elif len(str(stone)) % 2 == 0:
# Split the stone
str_stone = str(stone)
mid = len(str_stone) // 2
left = int(str_stone[:mid])
right = int(str_stone[mid:])
new_stones.extend([left, right])
else:
# Multiply by 2024
new_stones.append(stone * 2024)
return new_stones
def simulate_blinks(initial_stones, blinks):
stones = initial_stones
for _ in range(blinks):
stones = transform_stones(stones)
return stones
# Read input from file
file = "input.txt"
with open(file, "r") as f:
initial_stones = list(map(int, f.read().strip().split()))
# Part 1: After 25 blinks
stones_after_25_blinks = simulate_blinks(initial_stones, 25)
print(len(stones_after_25_blinks))
# Part 2: After 75 blinks
stones_after_75_blinks = simulate_blinks(initial_stones, 75)
print(len(stones_after_75_blinks))