2019 day 16 and 17 wip

2019/d17.py

@@ -0,0 +1,100 @@
+from lib import get_data, str_to_ints, Grid2D, add2
+from collections import defaultdict
+from d9 import Amp
+
+DIRS = [
+    (-1, 0),
+    (0, 1),
+    (1, 0),
+    (0, -1),
+]
+
+DIRCHAR = list("^>v<")
+
+
+def find_path(g):
+    (pos,) = g.find("^><v")
+    assert g[pos] == "^"
+
+    to_visit = set(g.find("#"))
+    seen = defaultdict(int)
+    dir = DIRS[0]  # up
+    seen[pos] = 1
+
+    def best_neighbor(pos):
+        best_nb = None
+        seen_count = 20
+        for nb in g.neighbors_ort(pos):
+            if nb in to_visit:
+                return nb
+            if nb in seen and seen[nb] < seen_count:
+                seen_count = seen[nb]
+                best_nb = nb
+        assert best_nb is not None
+        return best_nb
+
+    path = ""
+    while to_visit:
+
+        new_pos = add2(pos, dir)
+        if new_pos in to_visit or (new_pos in seen and seen[new_pos] < 2):
+            path += "F"
+            g[pos] = "#"
+            pos = new_pos
+            if pos in to_visit:
+                to_visit.remove(pos)
+            g[pos] = DIRCHAR[DIRS.index(dir)]
+            seen[pos] += 1
+        else:
+            best_nb = best_neighbor(pos)
+            while add2(pos, dir) != best_nb:
+                path += "R"
+                dir = DIRS[(DIRS.index(dir) + 1) % len(DIRS)]
+                g[pos] = DIRCHAR[DIRS.index(dir)]
+        g.print()
+        input()
+        print()
+    path = path.replace("RRR", "L")
+    return path
+
+
+def part_1(data):
+    xs = str_to_ints(data)
+    a = Amp(xs)
+    text = ""
+    while not a.done:
+        a.go()
+        while a.outputs:
+            o = a.pop()
+            text += chr(o)
+
+    result = 0
+    g = Grid2D(text)
+    for r in range(g.n_rows):
+        for c in range(g.n_cols):
+            nbs = g.neighbors_ort((r, c))
+            if (
+                g[(r, c)] == "#"
+                and len(nbs) == 4
+                and all([g[(nr, nc)] == "#" for nr, nc in nbs])
+            ):
+                # g[(r, c)] = 'o'
+                result += r * c
+    print(result)
+    g.print()
+    path = find_path(g)
+    print(path)
+
+    # xs = str_to_ints(data)
+    # xs[0] = 2
+    # a = Amp(xs)
+    # a.go()
+
+
+def main():
+    data = get_data(__file__)
+    part_1(data)
+
+
+if __name__ == "__main__":
+    main()