Moved 2022 to own dir

2023-11-30 19:20:34 +02:00
parent e700358bda
commit 6bd452a4ec
16 changed files with 0 additions and 0 deletions
--- a/2022/09/i.py
+++ b/2022/09/i.py
@@ -0,0 +1,46 @@
+import os
+
+input = open(os.path.dirname(__file__) +
+             "/input.txt", "r").readlines()
+
+
+def dir(d):
+    match d:
+        case 'R': return (1, 0)
+        case 'L': return (-1, 0)
+        case 'U': return (0, 1)
+        case 'D': return (0, -1)
+
+
+max = 10
+
+tails = []
+rope = [(0, 0) for i in range(max)]
+
+for l in input:
+    moves = l.split()
+    m = dir(moves[0])
+    for steps in range(int(moves[1])):
+        rope[0] = [rope[0][0]+m[0], rope[0][1]+m[1]]
+        for r in range(max-1):
+            k, kn = rope[r:r+2]
+            dx, dy = k[0]-kn[0], k[1] - kn[1]
+            dist = abs(dx) + abs(dy)
+            if dist == 2:
+                if k[0] == kn[0]:
+                    rope[r+1] = (rope[r+1][0], rope[r+1][1]+dy//2)
+                elif k[1] == kn[1]:
+                    rope[r+1] = (rope[r+1][0]+dx//2, rope[r+1][1])
+            elif dist == 3:
+                if abs(dx) == 2:
+                    rope[r+1] = (rope[r+1][0]+dx//2, rope[r][1])
+                elif abs(dy) == 2:
+                    rope[r+1] = (rope[r][0], rope[r+1][1]+dy//2)
+            elif dist == 4:
+                rope[r+1] = (rope[r+1][0]+dx//2, rope[r+1][1]+dy//2)
+        tails.append(rope[-1])
+
+l = []
+for i in tails:
+    l.append(str(i[0])+","+str(i[1]))
+print(len(set(l)))