[Python/2017] Move solutions into .py files

Python/2017/22.py

@@ -0,0 +1,58 @@
+from lib import *
+
+input = read_input(2017, 22)
+
+lines = input.splitlines()
+
+
+infected = set()
+x, y = len(lines[0]) // 2, len(lines) // 2
+dx, dy = 0, -1
+out = 0
+for i, line in enumerate(lines):
+    for j, c in enumerate(line):
+        if c == "#":
+            infected.add((j, i))
+
+for _ in range(10000):
+    inf = (x, y) in infected
+    if inf:
+        dx, dy = -dy, dx
+        infected.remove((x, y))
+    else:
+        dx, dy = dy, -dx
+        infected.add((x, y))
+        out += 1
+    x += dx
+    y += dy
+
+print(out)
+
+
+state = {}
+x, y = len(lines[0]) // 2, len(lines) // 2
+dx, dy = 0, -1
+out = 0
+for i, line in enumerate(lines):
+    for j, c in enumerate(line):
+        if c == "#":
+            state[(j, i)] = 2
+
+for _ in range(10000000):
+    s = state.get((x, y), 0)
+    if s == 0:
+        dx, dy = dy, -dx
+    elif s == 2:
+        dx, dy = -dy, dx
+    elif s == 3:
+        dx, dy = -dx, -dy
+    if s == 1:
+        out += 1
+    if s == 3:
+        state.pop((x, y))
+    else:
+        state[(x, y)] = s + 1
+    x += dx
+    y += dy
+
+print(out)