aoc 2017 (day 1-6)

www/statics/aoc/2017/06_solution-2.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+
+import aoc
+
+
+rawinput = aoc.read_input(6)
+
+mem = list(map(lambda x: int(x), rawinput.split('\t')))
+
+visited = dict()
+
+while True:
+    key = ";".join(map(lambda x: str(x), mem))
+    # print(mem)
+    if key in visited:
+        print(len(visited) - visited[key])
+        exit(0)
+    visited[key] = len(visited)
+
+    idx = mem.index(max(mem))
+    val = mem[idx]
+    mem[idx] = 0
+    for i in range(val):
+        mem[(idx+i+1) % len(mem)] += 1