[Python/2019] Restructure solutions

Python/2019/05.py

@@ -0,0 +1,83 @@
+from lib import *
+
+input = read_input(2019, 5)
+
+(*mem,) = map(int, input.split(","))
+pc = 0
+while pc < len(mem):
+    opcode = mem[pc] % 100
+    mode1 = mem[pc] // 100 % 10
+    mode2 = mem[pc] // 1000 % 10
+
+    if opcode == 1:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        mem[mem[pc + 3]] = arg1 + arg2
+        pc += 4
+    elif opcode == 2:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        mem[mem[pc + 3]] = arg1 * arg2
+        pc += 4
+    elif opcode == 3:
+        mem[mem[pc + 1]] = 1
+        pc += 2
+    elif opcode == 4:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        if arg1:
+            print(arg1)
+        pc += 2
+    elif opcode == 99:
+        break
+
+
+(*mem,) = map(int, input.split(","))
+pc = 0
+while pc < len(mem):
+    opcode = mem[pc] % 100
+    mode1 = mem[pc] // 100 % 10
+    mode2 = mem[pc] // 1000 % 10
+
+    if opcode == 1:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        mem[mem[pc + 3]] = arg1 + arg2
+        pc += 4
+    elif opcode == 2:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        mem[mem[pc + 3]] = arg1 * arg2
+        pc += 4
+    elif opcode == 3:
+        mem[mem[pc + 1]] = 5
+        pc += 2
+    elif opcode == 4:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        print(arg1)
+        pc += 2
+    elif opcode == 5:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        if arg1:
+            pc = arg2
+        else:
+            pc += 3
+    elif opcode == 6:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        if not arg1:
+            pc = arg2
+        else:
+            pc += 3
+    elif opcode == 7:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        mem[mem[pc + 3]] = int(arg1 < arg2)
+        pc += 4
+    elif opcode == 8:
+        arg1 = mem[pc + 1] if mode1 else mem[mem[pc + 1]]
+        arg2 = mem[pc + 2] if mode2 else mem[mem[pc + 2]]
+        mem[mem[pc + 3]] = int(arg1 == arg2)
+        pc += 4
+    elif opcode == 99:
+        break