summaryrefslogtreecommitdiffstats
path: root/19/py/d05.py
diff options
context:
space:
mode:
Diffstat (limited to '19/py/d05.py')
-rw-r--r--19/py/d05.py34
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))