Files
aocpy/2015/d3.py
2024-10-20 15:19:25 -04:00

37 lines
507 B
Python

from lib import get_data
from lib import V
data = get_data(__file__)
DIRS = {
"^": (-1, 0),
">": (0, 1),
"v": (1, 0),
"<": (0, -1),
}
pos = V(0, 0)
poss = set([pos])
for c in data:
d = DIRS[c]
pos = pos + d
poss.add(pos)
print(len(poss))
a = V(0, 0)
b = V(0, 0)
poss = set([a, b])
for i, c in enumerate(data):
if i % 2 == 0:
d = DIRS[c]
a = a + d
poss.add(a)
else:
d = DIRS[c]
b = b + d
poss.add(b)
print(len(poss))