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))