19.py

from lib import *

input = read_input(2019, 19)


class IntCode:
    def __init__(self, mem):
        self.mem = {i: e for i, e in enumerate(mem)}
        self.pc = 0
        self.running = False
        self.inp = []
        self.out = []
        self.rel = 0

    def start(self):
        self.running = True
        self.cont()

    def get_arg(self, i):
        mode = self.mem[self.pc] // (10 ** (i + 1)) % 10
        out = self.mem[self.pc + i]
        if mode == 1:
            return out
        elif mode == 2:
            out += self.rel
        return self.mem.get(out, 0)

    def write_arg(self, i, value):
        mode = self.mem[self.pc] // (10 ** (i + 1)) % 10
        pos = self.mem[self.pc + i]
        assert mode != 1
        if mode == 2:
            pos += self.rel
        self.mem[pos] = value

    def cont(self):
        while True:
            opcode = self.mem[self.pc] % 100

            if opcode == 1:
                self.write_arg(3, self.get_arg(1) + self.get_arg(2))
                self.pc += 4
            elif opcode == 2:
                self.write_arg(3, self.get_arg(1) * self.get_arg(2))
                self.pc += 4
            elif opcode == 3:
                if not self.inp:
                    return
                self.write_arg(1, self.inp.pop(0))
                self.pc += 2
            elif opcode == 4:
                self.out.append(self.get_arg(1))
                self.pc += 2
            elif opcode == 5:
                if self.get_arg(1):
                    self.pc = self.get_arg(2)
                else:
                    self.pc += 3
            elif opcode == 6:
                if not self.get_arg(1):
                    self.pc = self.get_arg(2)
                else:
                    self.pc += 3
            elif opcode == 7:
                self.write_arg(3, int(self.get_arg(1) < self.get_arg(2)))
                self.pc += 4
            elif opcode == 8:
                self.write_arg(3, int(self.get_arg(1) == self.get_arg(2)))
                self.pc += 4
            elif opcode == 9:
                self.rel += self.get_arg(1)
                self.pc += 2
            elif opcode == 99:
                self.running = False
                return


(*mem,) = map(int, input.split(","))


def test(x, y):
    intcode = IntCode(mem)
    intcode.inp += [x, y]
    intcode.start()
    return intcode.out.pop(0)


print(sum(test(i, j) for i in range(50) for j in range(50)))


(*mem,) = map(int, input.split(","))

dp = {}


def test(x, y):
    if (x, y) not in dp:
        intcode = IntCode(mem)
        intcode.inp += [x, y]
        intcode.start()
        dp[(x, y)] = intcode.out.pop(0)
    return dp[(x, y)]


SIZE = 100
x = 0
y = 0
done = False
while not done:
    y += 1
    for x in range(y + 1):
        if test(x, y - x):
            y = y - x
            done = True
            break


while True:
    while not test(x, y):
        y += 1
    while test(x, y + 1):
        y += 1

    assert test(x, y)
    if y >= SIZE - 1 and test(x + SIZE - 1, y - (SIZE - 1)):
        print(x * 10000 + y - (SIZE - 1))
        break

    x += 1