ollien/advent-of-code-2019
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Change execute_program to clone array inputs time in day2

Browse source
This commit is contained in:
Nick Krichevsky 2019-12-02 00:33:18 -05:00
parent 146c083627
commit d5a8214ad7
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
day2/main.py
View file

@ -6,7 +6,8 @@ def execute_program(inputs):
ADD_OPCODE = 1 ADD_OPCODE = 1
MULTIPLY_OPCODE = 2 MULTIPLY_OPCODE = 2
for i, opcode in itertools.islice(enumerate(inputs), 0, None, 4): program_memory = inputs[:]
for i, opcode in itertools.islice(enumerate(program_memory), 0, None, 4):
operation = None operation = None
if opcode == TERMINATE_OPCODE: if opcode == TERMINATE_OPCODE:
break break
@ -17,12 +18,12 @@ def execute_program(inputs):
else: else:
raise Exception('Bad opcode: ' + str(opcode)) raise Exception('Bad opcode: ' + str(opcode))
input_a_index = inputs[i + 1] input_a_index = program_memory[i + 1]
input_b_index = inputs[i + 2] input_b_index = program_memory[i + 2]
out_index = inputs[i + 3] out_index = program_memory[i + 3]
inputs[out_index] = operation(inputs[input_a_index], inputs[input_b_index]) program_memory[out_index] = operation(program_memory[input_a_index], program_memory[input_b_index])
return inputs[0] return program_memory[0]
def part1(inputs): def part1(inputs):
@ -40,7 +41,7 @@ def part2(inputs):
inputs[1] = i inputs[1] = i
inputs[2] = j inputs[2] = j
# Execute the program, making sure we clone the inputs array so we don't reuse the list # Execute the program, making sure we clone the inputs array so we don't reuse the list
result = execute_program(inputs[:]) result = execute_program(inputs)
if result == DESIRED_OUTPUT: if result == DESIRED_OUTPUT:
return 100 * i + j return 100 * i + j
else: else:
@ -51,5 +52,5 @@ if __name__ == '__main__':
with open('input.txt') as f: with open('input.txt') as f:
inputs = [int(item) for item in f.read().rstrip().split(',')] inputs = [int(item) for item in f.read().rstrip().split(',')]
print(part1(inputs[:])) print(part1(inputs))
print(part2(inputs[:])) print(part2(inputs))