diff options
Diffstat (limited to 'solutions/py')
| -rw-r--r-- | solutions/py/07.py | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/solutions/py/07.py b/solutions/py/07.py new file mode 100644 index 0000000..2fffe3f --- /dev/null +++ b/solutions/py/07.py @@ -0,0 +1,133 @@ +import itertools +import queue +import sys + +params = {1:3, 2:3, 3:1, 4:1, 5:2, 6:2, 7:3, 8:3, 99:0} +ADD = 1 +MULT = 2 +IN = 3 +OUT = 4 +JNZ = 5 +JEZ = 6 +LET = 7 +EQV = 8 +HAL = 99 + +class Computer(object): + def __init__(self, program): + self.memory = program.copy() + self.pointer = 0 + self.phase_read = False + + self.input = None + self.output = None + + def parse_op(self, op): + code = op % 100 + return [code] + [(op // 10**(i+2)) % 10**(i+1) for i in range(params[code])] + + def clear_flags(self): + self.input = None + self.output = None + + def get_param(self, inst, num): + return self.memory[self.pointer + num] if inst[num] == True else\ + self.memory[self.memory[self.pointer + num]] + + def step(self): + inst = self.parse_op(self.memory[self.pointer]) + if inst[0] == HAL: + return + elif inst[0] == ADD: + self.memory[self.memory[self.pointer+3]] = \ + self.get_param(inst, 1) + self.get_param(inst, 2) + self.pointer += 4 + elif inst[0] == MULT: + self.memory[self.memory[self.pointer+3]] = \ + self.get_param(inst, 1) * self.get_param(inst, 2) + self.pointer += 4 + elif inst[0] == IN: + self.memory[self.memory[self.pointer+1]] = self.input + self.input = None + self.pointer += 2 + elif inst[0] == OUT: + self.output = self.get_param(inst, 1) + self.pointer += 2 + elif inst[0] == JNZ: + if self.get_param(inst, 1) != 0: + self.pointer = self.get_param(inst, 2) + else: + self.pointer += 3 + elif inst[0] == JEZ: + if self.get_param(inst, 1) == 0: + self.pointer = self.get_param(inst, 2) + else: + self.pointer += 3 + elif inst[0] == LET: + self.memory[self.memory[self.pointer+3]] = 1 if \ + self.get_param(inst, 1) < self.get_param(inst, 2) \ + else 0 + self.pointer += 4 + elif inst[0] == EQV: + self.memory[self.memory[self.pointer+3]] = 1 if \ + self.get_param(inst, 1) == self.get_param(inst, 2) \ + else 0 + self.pointer += 4 + else: + print(self.memory) + print(self.pointer) + print("invalid instruction", self.memory[self.pointer]) + print(inst) + sys.exit() + +def pt2(program): + highest_signal = 0 + highest_sequence = None + for phase_seq in list(itertools.permutations(list(range(5,10)))): + signal = 0 + q = queue.Queue(5) + for phase in phase_seq: + q.put(phase) + amps = [Computer(program) for i in range(5)] + for amp in amps: + amp.input = q.get() + + signal = 0 + current_amp = 0 + while True: + amp = amps[current_amp] + amp.step() + if amp.input == None: + if amp.phase_read == False: + amp.phase_read = True + amp.input = signal + else: + pass + if amp.output is not None: + signal = amp.output + amp.output = None + current_amp = (current_amp + 1) % 5 + if amps[current_amp].phase_read == True: + amps[current_amp].input = signal + continue + if amp.memory[amp.pointer] == 99: + if current_amp == 4: + break + current_amp = (current_amp + 1) % 5 + amps[current_amp].input = signal + continue + if signal > highest_signal: + highest_signal = signal + highest_sequence = phase_seq + return highest_signal + +if __name__ == "__main__": + f = open("../input/07", "r") + program = [int(x) for x in f.readline().split(",")] + + import cProfile + import timeit + + print(pt2(program)) + print(timeit.timeit('pt2(program)', globals=globals(), number=1)*1000, "ms") + cProfile.run("pt2(program)") |