mirror of
https://github.com/peter-tanner/advent-of-code-2020.git
synced 2024-11-30 19:00:17 +08:00
52 lines
1.0 KiB
Python
52 lines
1.0 KiB
Python
|
|
||
|
|
# I love <3 complex numbers :3
|
||
|
|
|
||
|
|
import math
|
||
|
|
|
||
|
|
commands = [ [x[0],int(x[1:])] for x in open('input').read().splitlines()]
|
||
|
|
|
||
|
|
def pvec(d, m):
|
||
|
|
return {
|
||
|
|
'N' : complex(0,m),
|
||
|
|
'E' : complex(m,0),
|
||
|
|
'S' : complex(0,-m),
|
||
|
|
'W' : complex(-m,0)
|
||
|
|
}[d]
|
||
|
|
|
||
|
|
def a(commands):
|
||
|
|
fv = 0+0j
|
||
|
|
phase = 0
|
||
|
|
for v in commands:
|
||
|
|
d = v[0]
|
||
|
|
m = v[1]
|
||
|
|
if d == 'F':
|
||
|
|
fv += (m*(1j)**(phase/90))
|
||
|
|
elif d in ['L', 'R']:
|
||
|
|
phase += {
|
||
|
|
'L' : m,
|
||
|
|
'R' : -m
|
||
|
|
}[d]
|
||
|
|
else:
|
||
|
|
fv += pvec(d, m)
|
||
|
|
return (int)(abs(fv.real) + abs(fv.imag))
|
||
|
|
|
||
|
|
print(a(commands))
|
||
|
|
|
||
|
|
def b(commands):
|
||
|
|
fv = 0+0j
|
||
|
|
wv = 10+1j
|
||
|
|
for v in commands:
|
||
|
|
d = v[0]
|
||
|
|
m = v[1]
|
||
|
|
if d == 'F':
|
||
|
|
fv += m*wv
|
||
|
|
elif d in ['L', 'R']:
|
||
|
|
wv *= {
|
||
|
|
'L' : (1j)**(m/90),
|
||
|
|
'R' : (-1j)**(m/90)
|
||
|
|
}[d]
|
||
|
|
else:
|
||
|
|
wv += pvec(d, m)
|
||
|
|
return (int)(abs(fv.real) + abs(fv.imag))
|
||
|
|
|
||
|
|
print(b(commands))
|