From b904e20b104f043de7cf337e59ea2a44b52bccb0 Mon Sep 17 00:00:00 2001 From: Thibaud Gasser Date: Fri, 6 Dec 2019 17:49:34 +0100 Subject: [PATCH] [WIP] Day 5 challenge- Part 1: OK- Part 2: KO --- day5/day5.py | 123 +++++++++++++++++++++++++++++++++++++++++++++++++ day5/input.txt | 1 + 2 files changed, 124 insertions(+) create mode 100644 day5/day5.py create mode 100644 day5/input.txt diff --git a/day5/day5.py b/day5/day5.py new file mode 100644 index 0000000..5f24de0 --- /dev/null +++ b/day5/day5.py @@ -0,0 +1,123 @@ +#! /usr/bin/env python3 +from collections import namedtuple + + +def decode_opcode(inp): + opcode = inp % 100 # 2-digit opcode + a = (inp // 10000) % 10 # mode of 1st param + b = (inp // 1000) % 10 # mode of 2nd param + c = (inp // 100) % 10 # mode of 3rd param + # ABC DE + # ^^^ ^^ + # MODE OPCODE + return a, b, c, opcode + + +def decode_parameters(input_prog, ip, modes): + """ + Returns the adress of the parameters according to the input modes. + if mode[n] == 1: direct access, else indirect access + """ + param1 = input_prog[ip + 1] if modes[2] == 0 else (ip + 1) + param2 = input_prog[ip + 2] if modes[1] == 0 else (ip + 2) + param3 = input_prog[ip + 3] if modes[0] == 0 else (ip + 3) + return param1, param2, param3 + + + +def interpret_intcode(input_prog, stdin=None): + # Instruction pointer: index of the next element to be executed + ip = 0 + while ip < len(input_prog): + instruction = input_prog[ip] + *modes, opcode = decode_opcode(instruction) + + if opcode == 99: + break + + parameters = decode_parameters(input_prog, ip, modes) + + if opcode == 1: # ADD + print("ADD", " ".join(str(input_prog[p]) for p in parameters)) + input_prog[parameters[2]] = input_prog[parameters[0]] + input_prog[parameters[1]] + ip += 4 + + elif opcode == 2: # MUL + print("MUL", " ".join(str(input_prog[p]) for p in parameters)) + input_prog[parameters[2]] = input_prog[parameters[0]] * input_prog[parameters[1]] + ip += 4 + + elif opcode == 3: # IN + input_prog[parameters[1]] = stdin + ip += 2 + + elif opcode == 4: # OUT + if input_prog[ip + 1] == 99: + print("End of execution") + break + + print("OUT ", input_prog[parameters[1]]) + ip += 2 + + elif opcode == 5: # JMP if true + if input_prog[parameters[0]] != 0: + print("JT", " ".join(str(input_prog[p]) for p in parameters)) + ip = input_prog[parameters[1]] + else: + ip += 3 + + elif opcode == 6: # JMP if false + if input_prog[parameters[0]] == 0: + print("JF", " ".join(str(input_prog[p]) for p in parameters)) + ip = input_prog[parameters[1]] + else: + ip += 3 + + elif opcode == 7: # LT + print("LT", " ".join(str(input_prog[p]) for p in parameters)) + res = int(input_prog[parameters[0]] < input_prog[parameters[1]]) + input_prog[parameters[2]] = res + ip += 4 + + elif opcode == 8: # EQ + print("EQ", " ".join(str(input_prog[p]) for p in parameters)) + res = int(input_prog[parameters[1]] == input_prog[parameters[2]]) + input_prog[parameters[2]] = res + ip += 4 + + +def tests(): + inputs = ( + [1, 0, 0, 0, 99], # ADD 1 + 1 to location 0 + [2, 3, 0, 3, 99], # MUL 2 * 3 to location 3 + [2, 4, 4, 5, 99, 0], + [1, 1, 1, 4, 99, 5, 6, 0, 99], + ) + expected_outputs = ( + [2, 0, 0, 0, 99], # 1 + 1 = 2 + [2, 3, 0, 6, 99], # 3 * 2 = 6 + [2, 4, 4, 5, 99, 9801], # 99 * 99 = 9801 + [30, 1, 1, 4, 2, 5, 6, 0, 99], + ) + for i, inp in enumerate(inputs): + result = inp[:] + interpret_intcode(result) + assert ( + result == expected_outputs[i] + ), f"Expected output for {inp} is {expected_outputs[i]}, but found {result} instead." + print("\nAll tests passed.\n") + + +if __name__ == "__main__": + tests() + # a = [3, 12, 6, 12, 15, 1, 13, 14, 13, 4, 13, 99, -1, 0, 1, 9] + # b = [3, 3, 1105, -1, 9, 1101, 0, 0, 12, 4, 12, 99, 1] + + # interpret_intcode(b, 0) + # interpret_intcode(a, 0) + + with open("input.txt") as inp: + print("Start of input program.") + memory = [int(x) for x in inp.readline().strip().split(",")] + interpret_intcode(memory, 5) + diff --git a/day5/input.txt b/day5/input.txt new file mode 100644 index 0000000..425efbc --- /dev/null +++ b/day5/input.txt @@ -0,0 +1 @@ +3,225,1,225,6,6,1100,1,238,225,104,0,1102,67,92,225,1101,14,84,225,1002,217,69,224,101,-5175,224,224,4,224,102,8,223,223,101,2,224,224,1,224,223,223,1,214,95,224,101,-127,224,224,4,224,102,8,223,223,101,3,224,224,1,223,224,223,1101,8,41,225,2,17,91,224,1001,224,-518,224,4,224,1002,223,8,223,101,2,224,224,1,223,224,223,1101,37,27,225,1101,61,11,225,101,44,66,224,101,-85,224,224,4,224,1002,223,8,223,101,6,224,224,1,224,223,223,1102,7,32,224,101,-224,224,224,4,224,102,8,223,223,1001,224,6,224,1,224,223,223,1001,14,82,224,101,-174,224,224,4,224,102,8,223,223,101,7,224,224,1,223,224,223,102,65,210,224,101,-5525,224,224,4,224,102,8,223,223,101,3,224,224,1,224,223,223,1101,81,9,224,101,-90,224,224,4,224,102,8,223,223,1001,224,3,224,1,224,223,223,1101,71,85,225,1102,61,66,225,1102,75,53,225,4,223,99,0,0,0,677,0,0,0,0,0,0,0,0,0,0,0,1105,0,99999,1105,227,247,1105,1,99999,1005,227,99999,1005,0,256,1105,1,99999,1106,227,99999,1106,0,265,1105,1,99999,1006,0,99999,1006,227,274,1105,1,99999,1105,1,280,1105,1,99999,1,225,225,225,1101,294,0,0,105,1,0,1105,1,99999,1106,0,300,1105,1,99999,1,225,225,225,1101,314,0,0,106,0,0,1105,1,99999,8,226,226,224,102,2,223,223,1005,224,329,1001,223,1,223,1108,677,677,224,1002,223,2,223,1006,224,344,101,1,223,223,1007,226,677,224,102,2,223,223,1005,224,359,101,1,223,223,1007,677,677,224,1002,223,2,223,1006,224,374,101,1,223,223,1108,677,226,224,1002,223,2,223,1005,224,389,1001,223,1,223,108,226,677,224,102,2,223,223,1006,224,404,101,1,223,223,1108,226,677,224,102,2,223,223,1005,224,419,101,1,223,223,1008,677,677,224,102,2,223,223,1005,224,434,101,1,223,223,7,677,226,224,1002,223,2,223,1005,224,449,101,1,223,223,1008,226,226,224,102,2,223,223,1005,224,464,1001,223,1,223,107,226,677,224,1002,223,2,223,1006,224,479,1001,223,1,223,107,677,677,224,102,2,223,223,1005,224,494,1001,223,1,223,1008,226,677,224,102,2,223,223,1006,224,509,1001,223,1,223,1107,677,226,224,102,2,223,223,1005,224,524,101,1,223,223,1007,226,226,224,1002,223,2,223,1006,224,539,1001,223,1,223,107,226,226,224,102,2,223,223,1006,224,554,101,1,223,223,108,677,677,224,1002,223,2,223,1006,224,569,1001,223,1,223,7,226,677,224,102,2,223,223,1006,224,584,1001,223,1,223,8,677,226,224,102,2,223,223,1005,224,599,101,1,223,223,1107,677,677,224,1002,223,2,223,1005,224,614,101,1,223,223,8,226,677,224,102,2,223,223,1005,224,629,1001,223,1,223,7,226,226,224,1002,223,2,223,1006,224,644,1001,223,1,223,108,226,226,224,1002,223,2,223,1006,224,659,101,1,223,223,1107,226,677,224,1002,223,2,223,1006,224,674,101,1,223,223,4,223,99,226 \ No newline at end of file