09.py

from lib import *

input = read_input(2018, 9)


players, n = map(int, re.match(r"^(\d+) players; last marble is worth (\d+) points$", input).groups())
lst = deque([0])
scores = [0 for _ in range(players)]
for i in range(1, n + 1):
    p = i % players
    if i % 23:
        lst.rotate(-1)
        lst.append(i)
    else:
        lst.rotate(7)
        scores[p] += i + lst.pop()
        lst.rotate(-1)
print(max(scores))


players, n = map(int, re.match(r"^(\d+) players; last marble is worth (\d+) points$", input).groups())
n *= 100
lst = deque([0])
scores = [0 for _ in range(players)]
for i in range(1, n + 1):
    p = i % players
    if i % 23:
        lst.rotate(-1)
        lst.append(i)
    else:
        lst.rotate(7)
        scores[p] += i + lst.pop()
        lst.rotate(-1)
print(max(scores))