Files
advent-of-code/2015/day-3/part-1.py
2024-12-11 11:06:00 +01:00

32 lines
696 B
Python

import os
file_path = os.path.join(os.path.dirname(__file__), "./input.txt")
# file_path = os.path.join(os.path.dirname(__file__), "./test-input.txt")
input = open(file_path).read().strip()
def next_coordinates(current, symbol):
if symbol == "^":
return (current[0] - 1, current[1])
if symbol == "v":
return (current[0] + 1, current[1])
if symbol == ">":
return (current[0], current[1] + 1)
if symbol == "<":
return (current[0], current[1] - 1)
count = 0
visited = {(0, 0)}
x = 0
y = 0
for dir in list(input):
next = next_coordinates((x, y), dir)
x = next[0]
y = next[1]
visited.add(next)
print(next)
print(len(visited))