28 lines
772 B
Python
28 lines
772 B
Python
|
with open('day23-input', 'r') as file:
|
||
|
|
data = [l.strip('\n') for l in file]
|
||
|
|
|
||
|
|
tokens = [line.replace(',', '').split() for line in data]
|
||
|
|
|
||
|
|
def run_program(registers):
|
||
|
|
i = 0
|
||
|
|
while i < len(tokens):
|
||
|
|
line = tokens[i]
|
||
|
|
if line[0] == 'jmp':
|
||
|
|
i += int(line[1])
|
||
|
|
elif line[0] == 'jie':
|
||
|
|
i += int(line[2]) if registers[line[1]] % 2 == 0 else 1
|
||
|
|
elif line[0] == 'jio':
|
||
|
|
i += int(line[2]) if registers[line[1]] == 1 else 1
|
||
|
|
else:
|
||
|
|
i += 1
|
||
|
|
if line[0] == 'inc':
|
||
|
|
registers[line[1]] += 1
|
||
|
|
elif line[0] == 'hlf':
|
||
|
|
registers[line[1]] //= 2
|
||
|
|
elif line[0] == 'tpl':
|
||
|
|
registers[line[1]] *= 3
|
||
|
|
return registers
|
||
|
|
|
||
|
|
print(run_program({'a': 0, 'b': 0})) # b is Part 1
|
||
|
|
print(run_program({'a': 1, 'b': 0})) # b is Part 2
|