diff --git a/day12/part1.py b/day12/part1.py index 5e1f17f..2f5f6d7 100644 --- a/day12/part1.py +++ b/day12/part1.py @@ -3,12 +3,13 @@ lines = ((x[0], int(x[1:].strip())) for x in open("input.txt")) pos = 0 direction = 1 targets = {"E": 1, "N": 1j, "W": -1, "S": -1j} +angles = {"L": 1, "R": -1} for cmd, num in lines: if cmd in targets: pos += targets[cmd] * num - elif cmd in "LR": - times = (-1 if cmd == "R" else 1) * num / 90 + elif cmd in angles: + times = angles[cmd] * num / 90 direction = direction * 1j ** times else: pos += direction * num diff --git a/day12/part2.py b/day12/part2.py index 1d9ac1e..8381b0f 100644 --- a/day12/part2.py +++ b/day12/part2.py @@ -3,12 +3,13 @@ lines = ((x[0], int(x[1:].strip())) for x in open("input.txt")) pos = 0 direction = 10 + 1j targets = {"E": 1, "N": 1j, "W": -1, "S": -1j} +angles = {"L": 1, "R": -1} for cmd, num in lines: if cmd in targets: direction += targets[cmd] * num - elif cmd in "LR": - times = (-1 if cmd == "R" else 1) * num / 90 + elif cmd in angles: + times = angles[cmd] * num / 90 direction = direction * 1j ** times else: pos += direction * num