diff options
Diffstat (limited to '19/py/d05.py')
| -rw-r--r-- | 19/py/d05.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/19/py/d05.py b/19/py/d05.py new file mode 100644 index 0000000..1415b99 --- /dev/null +++ b/19/py/d05.py @@ -0,0 +1,34 @@ +import intcode + +def pt1(input): + program = [int(x) for x in input[0].split(",")] + c = intcode.Computer(program) + c.input = 1 + output = [] + while c.memory[c.pointer] != 99: + c.step() + if c.output != None: + output.append(c.output) + c.output = None + return output + +def pt2(input): + program = [int(x) for x in input[0].split(",")] + c = intcode.Computer(program) + c.input = 5 + output = [] + while c.memory[c.pointer] != 99: + c.step() + if c.output != None: + output.append(c.output) + c.output = None + return output + +if __name__ == "__main__": + import cProfile + + input = open("../input/05", "r").readlines() + cProfile.run("pt1(input)") + cProfile.run("pt2(input)") + print(pt1(input)) + print(pt2(input)) |
