🇵📝

y2022/p08.py

+from itertools import count, takewhile
+
+
+
+def heights(t, dt):
+    return takewhile(lambda h: h is not None, (grid.get(t + dt * i) for i in count(1)))
+
+
+def is_visible(t):
+    return any(grid[t] > max(heights(t, dt), default=-1) for dt in udlr)
+
+
+def score(t):
+    score = 1
+    for dt in udlr:
+        count = 0
+        for h in heights(t, dt):
+            count += 1
+            if h >= grid[t]:
+                break
+        score *= count
+    return score
+
+
+udlr = [-1j, 1j, -1, 1]
+text = open(0).read()
+grid = {complex(x, y): int(v) for y, line in enumerate(text.splitlines(), 1) for x, v in enumerate(line, 1)}
+
+ans1 = sum(is_visible(t) for t in grid)
+print(ans1)
+
+ans2 = max(score(t) for t in grid)
+print(ans2)