sebse/python-aoc-2020
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Day 12 (switched to dict.get())

Browse Source
This commit is contained in:
Sebastian Seedorf
2020-12-12 12:33:04 +01:00
parent f6fde17543
commit 4919e2fc92
2 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
day12/part1.py
View File

@@ -4,14 +4,12 @@ pos = 0
direction = 1 direction = 1
targets = {"E": 1, "N": 1j, "W": -1, "S": -1j} targets = {"E": 1, "N": 1j, "W": -1, "S": -1j}
angles = {"L": 1, "R": -1} angles = {"L": 1, "R": -1}
forward = {"F": 1}
for cmd, num in lines: for cmd, num in lines:
if cmd in targets: pos += targets.get(cmd, 0) * num
pos += targets[cmd] * num times = angles.get(cmd, 0) * num / 90
elif cmd in angles:
times = angles[cmd] * num / 90
direction = direction * 1j ** times direction = direction * 1j ** times
else: pos += forward.get(cmd, 0) * direction * num
pos += direction * num
print(int(abs(pos.real) + abs(pos.imag))) print(int(abs(pos.real) + abs(pos.imag)))

10
day12/part2.py
View File

@@ -4,14 +4,12 @@ pos = 0
direction = 10 + 1j direction = 10 + 1j
targets = {"E": 1, "N": 1j, "W": -1, "S": -1j} targets = {"E": 1, "N": 1j, "W": -1, "S": -1j}
angles = {"L": 1, "R": -1} angles = {"L": 1, "R": -1}
forward = {"F": 1}
for cmd, num in lines: for cmd, num in lines:
if cmd in targets: direction += targets.get(cmd, 0) * num
direction += targets[cmd] * num times = angles.get(cmd, 0) * num / 90
elif cmd in angles:
times = angles[cmd] * num / 90
direction = direction * 1j ** times direction = direction * 1j ** times
else: pos += forward.get(cmd, 0) * direction * num
pos += direction * num
print(int(abs(pos.real) + abs(pos.imag))) print(int(abs(pos.real) + abs(pos.imag)))