mirror of
https://github.com/thib8956/advent-of-code.git
synced 2025-07-01 14:25:48 +00:00
import 2019 aoc
This commit is contained in:
163
2019/day5/day5.py
Normal file
163
2019/day5/day5.py
Normal file
@ -0,0 +1,163 @@
|
||||
#! /usr/bin/env python3
|
||||
from collections import namedtuple
|
||||
from enum import Enum
|
||||
import logging
|
||||
|
||||
logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.WARN)
|
||||
|
||||
|
||||
def get_nth_digit(n, number):
|
||||
"Returns the nth digit of the input number"
|
||||
return number // 10 ** n % 10
|
||||
|
||||
|
||||
class Operation(Enum):
|
||||
ADDITION = 1
|
||||
MULTIPLICATION = 2
|
||||
INPUT = 3
|
||||
OUTPUT = 4
|
||||
JMP_IF_TRUE = 5
|
||||
JMP_IF_FALSE = 6
|
||||
LESS_THAN = 7
|
||||
EQUALS = 8
|
||||
TERMINATION = 99
|
||||
|
||||
|
||||
class Mode(Enum):
|
||||
POSITION = 0
|
||||
IMMEDIATE = 1
|
||||
|
||||
|
||||
class Instruction:
|
||||
def __init__(self, opcode):
|
||||
self.operation = Operation(opcode % 100)
|
||||
self.modes = [Mode(get_nth_digit(n, opcode)) for n in range(2, 5)]
|
||||
self.handler_name = f"handle_{self.operation.name.lower()}"
|
||||
self.handler = getattr(self, self.handler_name, self.handle_termination)
|
||||
|
||||
def __repr__(self):
|
||||
return f"Instruction({self.operation}, {self.modes})"
|
||||
|
||||
def handle(self, program, ip):
|
||||
return self.handler(program, ip)
|
||||
|
||||
def handle_addition(self, program, ip):
|
||||
first, second = self._get_parameters(program, ip)
|
||||
logging.debug(f"ADD {first} {second}")
|
||||
result = first + second
|
||||
# the last mode should *always* be POSITION
|
||||
program[program[ip + 3]] = result
|
||||
ip += 4
|
||||
return ip
|
||||
|
||||
def handle_multiplication(self, program, ip):
|
||||
first, second = self._get_parameters(program, ip)
|
||||
logging.debug(f"MUL {first} {second}")
|
||||
result = first * second
|
||||
# the last mode should *always* be POSITION
|
||||
program[program[ip + 3]] = result
|
||||
ip += 4
|
||||
return ip
|
||||
|
||||
def handle_input(self, program, ip):
|
||||
ip += 2
|
||||
program[program[ip + 1]] = int(input("Enter ID > "))
|
||||
return ip
|
||||
|
||||
def handle_output(self, program, ip):
|
||||
print("OUT ", program[program[ip + 1]])
|
||||
ip += 2
|
||||
return ip
|
||||
|
||||
def handle_jmp_if_true(self, program, ip):
|
||||
first, second = self._get_parameters(program, ip)
|
||||
logging.debug(f"JMPT {first} {second}")
|
||||
return second if first != 0 else ip + 3
|
||||
|
||||
def handle_jmp_if_false(self, program, ip):
|
||||
first, second = self._get_parameters(program, ip)
|
||||
logging.debug(f"JMPF {first} {second}")
|
||||
return second if first == 0 else ip + 3
|
||||
|
||||
def handle_less_than(self, program, ip):
|
||||
first, second = self._get_parameters(program, ip)
|
||||
logging.debug(f"LT {first} {second}")
|
||||
program[program[ip + 3]] = int(first < second)
|
||||
ip += 4
|
||||
return ip
|
||||
|
||||
def handle_equals(self, program, ip):
|
||||
first, second = self._get_parameters(program, ip)
|
||||
logging.debug(f"EQ {first} {second}")
|
||||
program[program[ip + 3]] = int(first == second)
|
||||
ip += 4
|
||||
return ip
|
||||
|
||||
def handle_termination(self, program, ip):
|
||||
print("HALT")
|
||||
return ip
|
||||
|
||||
def _get_parameters(self, program, ip):
|
||||
first = (
|
||||
program[ip + 1]
|
||||
if self.modes[0] is Mode.IMMEDIATE
|
||||
else program[program[ip + 1]]
|
||||
)
|
||||
second = (
|
||||
program[ip + 2]
|
||||
if self.modes[1] is Mode.IMMEDIATE
|
||||
else program[program[ip + 2]]
|
||||
)
|
||||
return first, second
|
||||
|
||||
|
||||
def interpret_intcode(program, stdin=None):
|
||||
ip = 0
|
||||
while program[ip] != 99:
|
||||
opcode = program[ip]
|
||||
instruction = Instruction(opcode)
|
||||
ip = instruction.handle(program, ip)
|
||||
|
||||
|
||||
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],
|
||||
[1101, 1, 1, 0, 99], # ADD 1 + 1 to location 0 (direct access)
|
||||
)
|
||||
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],
|
||||
[2, 1, 1, 0, 99], # 1 + 1 = 2 (direct access)
|
||||
)
|
||||
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")
|
||||
|
||||
|
||||
def run_input_program(filename):
|
||||
import os
|
||||
|
||||
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
|
||||
with open(os.path.join(SCRIPT_DIR, filename)) as inp:
|
||||
print("Start of input program.")
|
||||
memory = [int(x) for x in inp.readline().strip().split(",")]
|
||||
interpret_intcode(memory, 5)
|
||||
|
||||
|
||||
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)
|
||||
run_input_program("input.txt")
|
||||
192
2019/day5/day52.py
Normal file
192
2019/day5/day52.py
Normal file
@ -0,0 +1,192 @@
|
||||
import logging
|
||||
|
||||
logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.DEBUG)
|
||||
|
||||
|
||||
class Instruction:
|
||||
code, argument_num = 0, 0
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
pass
|
||||
|
||||
def new_pc(self, intcode):
|
||||
return intcode.pc + self.argument_num + 1
|
||||
|
||||
|
||||
class PlusInstruction(Instruction):
|
||||
code, argument_num = 1, 3
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
logging.debug(f"ADD {arguments[0].value} {arguments[1].value}")
|
||||
intcode.set(arguments[2].address, arguments[0].value + arguments[1].value)
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class MultiplyInstruction(Instruction):
|
||||
code, argument_num = 2, 3
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
logging.debug(f"MUL {arguments[0].value} {arguments[1].value}")
|
||||
intcode.set(arguments[2].address, arguments[0].value * arguments[1].value)
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class InputInstruction(Instruction):
|
||||
code, argument_num = 3, 1
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
intcode.set(arguments[0].address, intcode.get_input())
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class OutputInstruction(Instruction):
|
||||
code, argument_num = 4, 1
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
intcode.output = arguments[0].value
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class JumpIfTrueInstruction(Instruction):
|
||||
code, argument_num = 5, 2
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
logging.debug(f"JMPT {arguments[0].value} {arguments[1].value}")
|
||||
return arguments[1].value if arguments[0].value != 0 else self.new_pc(intcode)
|
||||
|
||||
|
||||
class JumpIfFalseInstruction(Instruction):
|
||||
code, argument_num = 6, 2
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
logging.debug(f"JMPF {arguments[0].value} {arguments[1].value}")
|
||||
return arguments[1].value if arguments[0].value == 0 else self.new_pc(intcode)
|
||||
|
||||
|
||||
class LessThanInstruction(Instruction):
|
||||
code, argument_num = 7, 3
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
logging.debug(f"LT {arguments[0].value} {arguments[1].value}")
|
||||
intcode.set(arguments[2].address, int(arguments[0].value < arguments[1].value))
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class EqualsInstruction(Instruction):
|
||||
code, argument_num = 8, 3
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
intcode.set(arguments[2].address, int(arguments[0].value == arguments[1].value))
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class RelativeBaseOffsetInstruction(Instruction):
|
||||
code, argument_num = 9, 1
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
intcode.relative_base += arguments[0].value
|
||||
return self.new_pc(intcode)
|
||||
|
||||
|
||||
class HaltInstruction(Instruction):
|
||||
code, argument_num = 99, 0
|
||||
|
||||
def execute(self, intcode, arguments):
|
||||
intcode.halted = True
|
||||
return None
|
||||
|
||||
|
||||
class Argument:
|
||||
def __init__(self, value, address):
|
||||
self.value = value
|
||||
self.address = address
|
||||
|
||||
|
||||
class IntCode:
|
||||
def __init__(self, program, inputs=[], input_func=None):
|
||||
self.program = program[:]
|
||||
self.inputs = inputs[::-1]
|
||||
self.input_func = input_func
|
||||
self.relative_base = 0
|
||||
self.memory = {}
|
||||
self.output = None
|
||||
self.halted = False
|
||||
self.pc = 0
|
||||
|
||||
def _get_instruction(self, instruction_code):
|
||||
return next(
|
||||
cls for cls in Instruction.__subclasses__() if cls.code == instruction_code
|
||||
)
|
||||
|
||||
def _parse_arguments(self, argument_num):
|
||||
modes = str(self.program[self.pc]).zfill(5)[:3][::-1]
|
||||
arguments = []
|
||||
for i in range(argument_num):
|
||||
value = self.program[self.pc + i + 1] + (
|
||||
self.relative_base if modes[i] == "2" else 0
|
||||
)
|
||||
if modes[i] == "1":
|
||||
arguments.append(Argument(value, value))
|
||||
else:
|
||||
arguments.append(
|
||||
Argument(
|
||||
self.program[value]
|
||||
if value < len(self.program)
|
||||
else self.memory.get(value, 0),
|
||||
value,
|
||||
)
|
||||
)
|
||||
return arguments
|
||||
|
||||
def run(self):
|
||||
self.output = None
|
||||
while not self.halted and self.output is None:
|
||||
instruction = self._get_instruction(self.program[self.pc] % 100)
|
||||
arguments = self._parse_arguments(instruction.argument_num)
|
||||
self.pc = instruction().execute(self, arguments)
|
||||
return self.output
|
||||
|
||||
def execute(self):
|
||||
last_output = None
|
||||
while not self.halted:
|
||||
output = self.run()
|
||||
if not self.halted:
|
||||
last_output = output
|
||||
return last_output
|
||||
|
||||
def clone(self):
|
||||
cloned = IntCode(self.program)
|
||||
cloned.inputs = self.inputs[:]
|
||||
cloned.input_func = self.input_func
|
||||
cloned.relative_base = self.relative_base
|
||||
cloned.memory = {key: value for key, value in self.memory.items()}
|
||||
cloned.output = self.output
|
||||
cloned.halted = self.halted
|
||||
cloned.pc = self.pc
|
||||
return cloned
|
||||
|
||||
def get(self, address):
|
||||
return (
|
||||
self.program[address]
|
||||
if address < len(self.program)
|
||||
else self.memory.get(value + self.relative_base, 0)
|
||||
)
|
||||
|
||||
def set(self, address, value):
|
||||
target = self.program if address < len(self.program) else self.memory
|
||||
target[address] = value
|
||||
|
||||
def input(self, value):
|
||||
self.inputs = [value] + self.inputs
|
||||
|
||||
def get_input(self):
|
||||
return self.inputs.pop() if self.inputs else self.input_func()
|
||||
|
||||
|
||||
import os
|
||||
|
||||
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
|
||||
code = list(map(int, open(os.path.join(SCRIPT_DIR, "input.txt")).read().split(",")))
|
||||
|
||||
part2 = IntCode(code, [5]).execute()
|
||||
print(part2)
|
||||
1
2019/day5/input.txt
Normal file
1
2019/day5/input.txt
Normal file
@ -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
|
||||
Reference in New Issue
Block a user