2020/12

y2020/p12.py

+# flake8: noqa
+import sys
+
+
+text = sys.stdin.read()
+
+
+pos = 0
+aim = 1
+for line in text.splitlines():
+    char, step = line[0], int(line[1:])
+    if char == 'N': pos += step * 1j
+    elif char == 'S': pos -= step * 1j
+    elif char == 'E': pos += step
+    elif char == 'W': pos -= step
+    elif char == 'F': pos += step * aim
+    elif char == 'L': aim *= 1j ** (step // 90)
+    elif char == 'R': aim /= 1j ** (step // 90)
+print(abs(pos.real) + abs(pos.imag))
+
+
+pos = 0
+way = 10 + 1j
+for line in text.splitlines():
+    char, step = line[0], int(line[1:])
+    if char == 'N': way += step * 1j
+    elif char == 'S': way -= step * 1j
+    elif char == 'E': way += step
+    elif char == 'W': way -= step
+    elif char == 'F': pos += step * way
+    elif char == 'L': way *= 1j ** (step // 90)
+    elif char == 'R': way /= 1j ** (step // 90)
+print(abs(pos.real) + abs(pos.imag))