1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
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))
|
