1 files changed, 67 insertions, 0 deletions
diff --git a/2024/day15/solve.py b/2024/day15/solve.py
new file mode 100644
index 0000000..5459df4
--- /dev/null
+++ b/2024/day15/solve.py
@@ -0,0 +1,67 @@
+def move(data: list[list[str]], current_pos: tuple[int, int],
+         current_order: str, movement_dict: dict[str, tuple[int, int]]) -> tuple[int, int]:
+    assert data[current_pos[0]][current_pos[1]] == "@"
+
+    direction = movement_dict[current_order]
+    can_move = True
+    steps = 0
+    while True:
+        steps += 1
+        current_tile = data[current_pos[0] + direction[0] * steps][current_pos[1] + direction[1] * steps]
+        if current_tile == "#":
+            can_move = False
+            break
+        elif current_tile == ".":
+            break
+
+    if can_move:
+        data[current_pos[0]][current_pos[1]] = "."
+        if steps > 1:
+            data[current_pos[0] + direction[0] * steps][current_pos[1] + direction[1] * steps] = "O"
+        current_pos = (current_pos[0] + direction[0], current_pos[1] + direction[1])
+        data[current_pos[0]][current_pos[1]] = "@"
+
+    return current_pos
+
+def calculate_gps_data(data: list[list[str]]):
+    result = 0
+    for row_pos, row in enumerate(data):
+        for col_pos, char in enumerate(row):
+            if char == "O":
+                result += row_pos * 100 + col_pos
+    return result
+
+movement_dirs = { "<": (0, -1), "v": (1, 0), "^": (-1, 0), ">": (0, 1) }
+map_data = []
+movement_data = ""
+
+with open("input") as f:
+    reading_movement = False
+    for line in f:
+        line = line.strip()
+        if reading_movement:
+            movement_data += line
+        else:
+            if line == "":
+                reading_movement = True
+            else:
+                map_data.append(list(line))
+
+for row_idx, row in enumerate(map_data):
+    for col_idx, char in enumerate(row):
+        if char == "@":
+            robot_row = row_idx
+            robot_col = col_idx
+            break
+    else:
+        continue
+    break
+
+new_pos = (robot_row, robot_col)
+for order in movement_data:
+    new_pos = move(map_data,new_pos, order, movement_dirs)
+
+for line in map_data:
+    print("".join(line))
+
+print(calculate_gps_data(map_data))
+\ No newline at end of file