42 lines
628 B
Python
42 lines
628 B
Python
import sys
|
|
from lib import *
|
|
|
|
data = open(0).read()
|
|
|
|
part_1 = False
|
|
|
|
DIRS = {
|
|
"^": (-1, 0),
|
|
">": (0, 1),
|
|
"v": (1, 0),
|
|
"<": (0, -1),
|
|
}
|
|
|
|
if part_1:
|
|
pos = (0, 0)
|
|
poss = set([pos])
|
|
|
|
for c in data:
|
|
d = DIRS[c]
|
|
pos = pos[0] + d[0], pos[1] + d[1]
|
|
poss.add(pos)
|
|
|
|
print(len(poss))
|
|
sys.exit(0)
|
|
|
|
a = (0, 0)
|
|
b = (0, 0)
|
|
poss = set([a, b])
|
|
for i, c in enumerate(data):
|
|
if i % 2 == 0:
|
|
d = DIRS[c]
|
|
a = a[0] + d[0], a[1] + d[1]
|
|
poss.add(a)
|
|
else:
|
|
d = DIRS[c]
|
|
b = b[0] + d[0], b[1] + d[1]
|
|
poss.add(b)
|
|
|
|
print(len(poss))
|
|
sys.exit(0)
|