27 lines
890 B
Python
27 lines
890 B
Python
|
import numpy as np
|
||
|
|
from itertools import cycle
|
||
|
|
from blist import blist # Regular lists are far too slow for Part 2, as are numpy arrays.
|
||
|
|
|
||
|
|
with open('day9-input', 'r') as file:
|
||
|
|
data = [l.strip('\n') for l in file][0].split()
|
||
|
|
players = int(data[0])
|
||
|
|
last_marble = int(data[-2])
|
||
|
|
|
||
|
|
|
||
|
|
def find_winning_score(players, last_marble):
|
||
|
|
marbles = blist([0])
|
||
|
|
scores = np.zeros(players, dtype=np.int64)
|
||
|
|
current_marble = 0
|
||
|
|
for marble, player in zip(range(1, last_marble+1), cycle(range(players))):
|
||
|
|
if marble % 23 == 0:
|
||
|
|
current_marble = (current_marble-7) % len(marbles)
|
||
|
|
scores[player] += marble + marbles.pop(current_marble)
|
||
|
|
else:
|
||
|
|
current_marble = (current_marble+2) % len(marbles)
|
||
|
|
marbles.insert(current_marble, marble)
|
||
|
|
return max(scores)
|
||
|
|
|
||
|
|
|
||
|
|
print(find_winning_score(players, last_marble)) # Part 1
|
||
|
|
print(find_winning_score(players, last_marble*100)) # Part 2
|