mirror of
https://github.com/ershisan99/advent-of-code.git
synced 2025-12-16 12:32:49 +00:00
50 lines
938 B
Python
50 lines
938 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)}
|
|
|
|
santa_queue = []
|
|
robot_queue = []
|
|
|
|
for i, foo in enumerate(list(input)):
|
|
if i % 2 == 0:
|
|
santa_queue.append(foo)
|
|
else:
|
|
robot_queue.append(foo)
|
|
|
|
|
|
def deliver(queue):
|
|
x = 0
|
|
y = 0
|
|
|
|
for dir in queue:
|
|
next = next_coordinates((x, y), dir)
|
|
visited.add(next)
|
|
|
|
x = next[0]
|
|
y = next[1]
|
|
|
|
|
|
deliver(santa_queue)
|
|
deliver(robot_queue)
|
|
|
|
|
|
print(len(visited))
|