Added day 16HEADmaster

1 files changed, 2 insertions, 6 deletions

diff --git a/day15/__init__.py b/day15/__init__.py
index 38126fc..9eb94d9 100644
--- a/day15/__init__.py
+++ b/day15/__init__.py
@@ -116,17 +116,13 @@ class AssignmentOne(Assignment):
     example_result = 40
 
 class AssignmentTwo(Assignment):
-    example_result =  315
-
-    @classmethod
-    def overflow(self, item):
-        return item - 9 if item > 9 else item
+    example_result = 315
 
     @classmethod
     def increase_map(cls, map) -> Map:
         return [
             [
-                cls.overflow(item + dx + dy)
+                ((item + dx + dy - 1) % 9) + 1
                 for dx in range(5)
                 for item in row
             ]