Author: Honza Pokorny <me@honza.ca>
Add part 1 of day 5
005/main.py | 114 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
diff --git a/005/main.py b/005/main.py new file mode 100644 index 0000000000000000000000000000000000000000..0ed03790528d657ec52d2c1ad24fd1f01c7e0feb --- /dev/null +++ b/005/main.py @@ -0,0 +1,114 @@ +import re +import copy + +opcode = re.compile(r'(\d{2})') + +OPS = ['01', '02', '03', '04', '99'] + + +def rev(s): + r = [] + for l in s: + r.append(l) + + r.reverse() + return ''.join(r) + + + +class NonHaltException(Exception): + pass + + +def add(a, b): + return a + b + + +def mult(a, b): + return a * b + + +def intcode(input_list, input_code): + cur = 0 + if input_list[0] == 3: + input_list[input_list[1]] = input_code + cur += 2 + + while True: + op_value = input_list[cur] + op_value_string = '000000' + str(op_value) + + op = op_value_string[-2:] + + front = op_value_string[0:-2] + front = rev(front) + + if op == '99': + break + + if op == '01': + # 0 po, like before + # 1 im + a_mode = front[0] + b_mode = front[1] + + if a_mode == '0': + a = input_list[input_list[cur + 1]] + else: + a = input_list[cur + 1] + + if b_mode == '0': + b = input_list[input_list[cur + 2]] + else: + b = input_list[cur + 2] + + result_pos = input_list[cur + 3] + input_list[result_pos] = add(a, b) + cur += 4 + continue + + if op == '02': + # 0 po, like before + # 1 im + a_mode = front[0] + b_mode = front[1] + + if a_mode == '0': + a = input_list[input_list[cur + 1]] + else: + a = input_list[cur + 1] + + if b_mode == '0': + b = input_list[input_list[cur + 2]] + else: + b = input_list[cur + 2] + + result_pos = input_list[cur + 3] + input_list[result_pos] = mult(a, b) + cur += 4 + continue + + if op == '04': + mode = front[0] + if mode == '0': + print(input_list[input_list[cur + 1]]) + else: + print(input_list[cur + 1]) + cur += 2 + continue + + +def main(): + input_list = [ + + 3,225,1,225,6,6,1100,1,238,225,104,0,1102,45,16,225,2,65,191,224,1001,224,-3172,224,4,224,102,8,223,223,1001,224,5,224,1,223,224,223,1102,90,55,225,101,77,143,224,101,-127,224,224,4,224,102,8,223,223,1001,224,7,224,1,223,224,223,1102,52,6,225,1101,65,90,225,1102,75,58,225,1102,53,17,224,1001,224,-901,224,4,224,1002,223,8,223,1001,224,3,224,1,224,223,223,1002,69,79,224,1001,224,-5135,224,4,224,1002,223,8,223,1001,224,5,224,1,224,223,223,102,48,40,224,1001,224,-2640,224,4,224,102,8,223,223,1001,224,1,224,1,224,223,223,1101,50,22,225,1001,218,29,224,101,-119,224,224,4,224,102,8,223,223,1001,224,2,224,1,223,224,223,1101,48,19,224,1001,224,-67,224,4,224,102,8,223,223,1001,224,6,224,1,223,224,223,1101,61,77,225,1,13,74,224,1001,224,-103,224,4,224,1002,223,8,223,101,3,224,224,1,224,223,223,1102,28,90,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,7,226,677,224,102,2,223,223,1005,224,329,1001,223,1,223,8,226,677,224,1002,223,2,223,1005,224,344,101,1,223,223,8,226,226,224,1002,223,2,223,1006,224,359,101,1,223,223,1008,677,226,224,1002,223,2,223,1005,224,374,1001,223,1,223,108,677,677,224,1002,223,2,223,1005,224,389,1001,223,1,223,1107,226,677,224,1002,223,2,223,1006,224,404,101,1,223,223,1008,226,226,224,102,2,223,223,1006,224,419,1001,223,1,223,7,677,226,224,1002,223,2,223,1005,224,434,101,1,223,223,1108,226,226,224,1002,223,2,223,1005,224,449,101,1,223,223,7,226,226,224,102,2,223,223,1005,224,464,101,1,223,223,108,677,226,224,102,2,223,223,1005,224,479,1001,223,1,223,1007,677,226,224,1002,223,2,223,1006,224,494,1001,223,1,223,1007,677,677,224,1002,223,2,223,1006,224,509,1001,223,1,223,107,677,677,224,1002,223,2,223,1005,224,524,101,1,223,223,1108,226,677,224,102,2,223,223,1006,224,539,1001,223,1,223,8,677,226,224,102,2,223,223,1005,224,554,101,1,223,223,1007,226,226,224,102,2,223,223,1006,224,569,1001,223,1,223,107,677,226,224,102,2,223,223,1005,224,584,1001,223,1,223,108,226,226,224,102,2,223,223,1006,224,599,1001,223,1,223,107,226,226,224,1002,223,2,223,1006,224,614,1001,223,1,223,1108,677,226,224,1002,223,2,223,1005,224,629,1001,223,1,223,1107,677,677,224,102,2,223,223,1005,224,644,1001,223,1,223,1008,677,677,224,102,2,223,223,1005,224,659,101,1,223,223,1107,677,226,224,1002,223,2,223,1006,224,674,101,1,223,223,4,223,99,226 + + + ] + + intcode(input_list, 1) + + + +if __name__ == '__main__': + main()