Day 12 (rename drctn to direction)

This commit is contained in:
Sebastian Seedorf
2020-12-12 12:13:37 +01:00
parent af4e4f012e
commit 801ac37dde
2 changed files with 7 additions and 7 deletions

View File

@@ -1,7 +1,7 @@
lines = ((x[0], int(x[1:].strip())) for x in open("input.txt"))
pos = 0
drctn = 1
direction = 1
targets = {"E": 1, "N": 1j, "W": -1, "S": -1j}
for cmd, num in lines:
@@ -9,8 +9,8 @@ for cmd, num in lines:
pos += targets[cmd] * num
elif cmd in "LR":
times = (-1 if cmd == "R" else 1) * num / 90
drctn = drctn * 1j ** times
direction = direction * 1j ** times
else:
pos += drctn * num
pos += direction * num
print(int(abs(pos.real) + abs(pos.imag)))

View File

@@ -1,16 +1,16 @@
lines = ((x[0], int(x[1:].strip())) for x in open("input.txt"))
pos = 0
drctn = 10+1j
direction = 10 + 1j
targets = {"E": 1, "N": 1j, "W": -1, "S": -1j}
for cmd, num in lines:
if cmd in targets:
drctn += targets[cmd] * num
direction += targets[cmd] * num
elif cmd in "LR":
times = (-1 if cmd == "R" else 1) * num / 90
drctn = drctn * 1j ** times
direction = direction * 1j ** times
else:
pos += drctn * num
pos += direction * num
print(int(abs(pos.real) + abs(pos.imag)))