X-Git-Url: https://git.sommitrealweird.co.uk/advent-of-code-2020.git/blobdiff_plain/11d744f1cd2c87d7dc01c504e368d4cecd99aa47..a3646324ce6a3441425f24c318df246b610f529d:/day15/memory.py

diff --git a/day15/memory.py b/day15/memory.py
new file mode 100644
index 0000000..2625a1e
--- /dev/null
+++ b/day15/memory.py
@@ -0,0 +1,24 @@
+#!/usr/bin/python3
+
+from collections import deque
+
+def get_answer(initial_data, end_count):
+    x=len(initial_data)
+    seen={}
+    for index,number in enumerate(initial_data,1):
+        seen[number]=index
+    last_answer=initial_data[-1]
+    while x < end_count:
+        answer=0
+        if last_answer in seen:
+            answer=x-seen[last_answer]
+        seen[last_answer]=x
+        x+=1
+        last_answer=answer
+        yield answer
+
+
+initial_data=[int(a) for a in open("input.txt", "r").readline().split(",")]
+
+print("Part 1: ", deque((get_answer(initial_data,2020)),maxlen=1).pop())
+print("Part 2: ", deque((get_answer(initial_data,30000000)),maxlen=1).pop())