Deque version of 2018 Day 9.

2018/day9-deque.py

@@ -0,0 +1,25 @@
+import numpy as np
+from itertools import cycle
+from collections import deque
+
+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 = deque([0])
+  scores = np.zeros(players, dtype=np.int64)
+  for marble, player in zip(range(1, last_marble+1), cycle(range(players))):
+    if marble % 23 == 0:
+      marbles.rotate(-7)
+      scores[player] += marble + marbles.pop()
+    else:
+      marbles.rotate(2)
+      marbles.append(marble)
+  return max(scores)
+
+
+print(find_winning_score(players, last_marble))  # Part 1
+print(find_winning_score(players, last_marble*100))  # Part 2