2017 day 22 part 1

2017/22/virus.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+import sys
+from collections import defaultdict
+data = [l.strip() for l in sys.stdin.readlines()]
+grid = defaultdict(bool)
+for i, row in enumerate(data):
+    for j, col in enumerate(row):
+        grid[complex(j, i)] = (col == "#")
+pos, d = complex(len(data) // 2, len(data[0]) // 2), 1j
+becomeinfected = 0
+for _ in range(10000):
+    if grid[pos]:
+        d *= 1j
+    else:
+        d *= -1j
+        becomeinfected += 1
+    grid[pos] = not grid[pos]
+    pos += d
+print(becomeinfected)

README.md

@@ -14,7 +14,7 @@ My solutions to the Advent of Code puzzles.
  6 ██    ██       ██
  7 ██    ██
  8 ██    ██
- 9 █     ██
+ 9       ██
 10 ██    ██
 11 ██    ██
 12 ██    ██