Added day 9

4 files changed, 1098 insertions, 2 deletions

diff --git a/aoc/__main__.py b/aoc/__main__.py
index 8c6a21e..efb215d 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -2,6 +2,8 @@ import argparse
 import os
 import sys
 import importlib
+from typing import Tuple, List
+
 
 def day(assignment_part: str) -> str:
     return {
@@ -9,6 +11,9 @@ def day(assignment_part: str) -> str:
         '2': 'Two',
     }[assignment_part]
 
+def kwargs(kwarg: str) -> List:
+    return kwarg.split('=')
+
 parser = argparse.ArgumentParser(description='Advent of Code')
 
 parser.add_argument('day', type=str, help='Assignment day')
@@ -17,6 +22,7 @@ parser.add_argument(
     help='Assingment part. Defaults to one.'
 )
 parser.add_argument('--example', default=False, action='store_true')
+parser.add_argument('kwargs', type=kwargs, nargs='*')
 
 
 if __name__ == '__main__':
@@ -24,6 +30,13 @@ if __name__ == '__main__':
     assignment_day = importlib.import_module(args.day)
 
     Assignment = getattr(assignment_day, f'Assignment{args.part}')
-    assignment = Assignment(path=os.path.dirname(assignment_day.__file__))
+    assignment = Assignment(
+        path=os.path.dirname(assignment_day.__file__),
+        **dict(args.kwargs)
+    )
 
-    print(assignment.run(input=assignment.read_input(example=args.example)))
+    print(
+        assignment.run(
+            input=assignment.read_input(example=args.example),
+        )
+    )
diff --git a/day9/__init__.py b/day9/__init__.py
new file mode 100644
index 0000000..0cf3770
--- /dev/null
+++ b/day9/__init__.py
@@ -0,0 +1,63 @@
+from typing import Any, Iterator, List
+
+from aoc import BaseAssignment
+
+
+class Assignment(BaseAssignment):
+    def __init__(self, path, preamble_length=25):
+        super().__init__(path)
+        self.preamble_length = int(preamble_length)
+
+    def read_input(self, example=False) -> List:
+        return list(super(Assignment, self).read_input(example))
+
+    def parse_item(self, item: str) -> Any:
+        return int(item)
+
+
+class AssignmentOne(Assignment):
+    def find_invalid(self, input: List, preamble_length: int = 25) -> int:
+        preamble = sorted(input[:preamble_length])
+        to_check = input[preamble_length]
+
+        front_position = 0
+        end_position = preamble_length - 1
+
+        while True:
+            if front_position is end_position:
+                return to_check
+
+            sum = preamble[front_position] + preamble[end_position]
+
+            if sum > to_check:
+                end_position -= 1
+            elif sum < to_check:
+                front_position += 1
+            else:
+                break
+
+        return self.find_invalid(input[1:], preamble_length=preamble_length)
+
+    def run(self, input: List) -> int:
+        return self.find_invalid(
+            input, preamble_length=int(self.preamble_length))
+
+
+class AssignmentTwo(AssignmentOne):
+    def find_weakness(self, input: List, invalid_nr: int) -> int:
+        for index, current_num in enumerate(input):
+            if index < 2:
+                continue
+
+            items_to_check = input[:index]
+            sum_list = sum(items_to_check)
+
+            if sum_list > invalid_nr:
+                break
+            elif sum_list == invalid_nr:
+                return min(items_to_check) + max(items_to_check)
+
+        return self.find_weakness(input[1:], invalid_nr)
+
+    def run(self, input: List) -> int:
+        return self.find_weakness(input, super().run(input))
diff --git a/day9/example.txt b/day9/example.txt
new file mode 100644
index 0000000..cda4246
--- /dev/null
+++ b/day9/example.txt
@@ -0,0 +1,20 @@
+35
+20
+15
+25
+47
+40
+62
+55
+65
+95
+102
+117
+150
+182
+127
+219
+299
+277
+309
+576
\ No newline at end of file
diff --git a/day9/input.txt b/day9/input.txt
new file mode 100644
index 0000000..e9932d5
--- /dev/null
+++ b/day9/input.txt
@@ -0,0 +1,1000 @@
+18
+19
+46
+14
+29
+45
+40
+47
+25
+43
+36
+22
+21
+4
+32
+33
+37
+38
+26
+2
+42
+15
+5
+13
+31
+9
+6
+30
+7
+14
+10
+8
+69
+11
+12
+25
+16
+17
+22
+19
+18
+20
+51
+21
+24
+23
+26
+15
+60
+13
+29
+27
+44
+42
+28
+38
+30
+36
+31
+32
+33
+34
+35
+37
+55
+39
+40
+70
+41
+43
+78
+58
+73
+101
+60
+56
+57
+75
+59
+67
+71
+61
+68
+110
+69
+80
+72
+112
+76
+79
+144
+129
+114
+84
+99
+113
+132
+115
+148
+116
+118
+120
+128
+126
+187
+130
+190
+152
+229
+141
+151
+155
+299