Spaces:
Running
Running
File size: 2,237 Bytes
a4da721 |
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 |
def solve():
keypad = ["789", "456", "123", " 0A"]
keypad_map = {}
for r in range(len(keypad)):
for c in range(len(keypad[r])):
keypad_map[keypad[r][c]] = (r, c)
directional_keypad = [" ^A", "<v>", " ^A", "<v>"]
directional_keypad_map = {}
for r in range(len(directional_keypad)):
for c in range(len(directional_keypad[r])):
directional_keypad_map[directional_keypad[r][c]] = (r, c)
def get_code(instructions, start_pos, keypad_layout, keypad_mapping):
code = ""
curr_pos = start_pos
for instruction in instructions:
next_pos = (curr_pos[0] + (1 if instruction == 'v' else (-1 if instruction == '^' else 0)),
curr_pos[1] + (1 if instruction == '>' else (-1 if instruction == '<' else 0)))
if 0 <= next_pos[0] < len(keypad_layout) and 0 <= next_pos[1] < len(keypad_layout[next_pos[0]]) and keypad_layout[next_pos[0]][next_pos[1]] != ' ':
curr_pos = next_pos
if instruction == 'A':
code += keypad_layout[curr_pos[0]][curr_pos[1]]
return code
def part1(instructions_list):
total_complexity = 0
for instructions in instructions_list:
code = get_code(instructions, keypad_map['A'], keypad, keypad_map)
complexity = len(instructions) * int(code[:-1] or "0")
total_complexity += complexity
return total_complexity
def part2(instructions_list):
total_complexity = 0
for instructions in instructions_list:
intermediate_instructions = instructions
for _ in range(25):
intermediate_instructions = get_code(intermediate_instructions, directional_keypad_map['A'], directional_keypad, directional_keypad_map)
code = get_code(intermediate_instructions, keypad_map['A'], keypad, keypad_map)
complexity = len(instructions) * int(code[:-1] or "0")
total_complexity += complexity
return total_complexity
with open("./input.txt") as f:
instructions_list = [line.strip() for line in f]
print(part1(instructions_list))
print(part2(instructions_list))
solve() |