From 63eafb5f424f9f7d84ca9ccdff93bf72c7f1aab6 Mon Sep 17 00:00:00 2001
From: Stefan Harmuth <stefan.harmuth@net-m.de>
Date: Tue, 29 Nov 2022 19:35:27 +0100
Subject: [PATCH] code cleanup

---
 day01.py | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/day01.py b/day01.py
index 7bd619d..e705b8a 100644
--- a/day01.py
+++ b/day01.py
@@ -13,24 +13,20 @@ class Day(AOCDay):
         ]
     ]
 
-    def part1(self):
+    def count(self, step: int = 1) -> int:
         count = 0
         depths = self.getInputListAsType(int)
-        for x in range(1, len(depths)):
-            if depths[x] > depths[x-1]:
+        for x in range(step, len(depths)):
+            if depths[x] > depths[x - step]:
                 count += 1
 
         return count
 
+    def part1(self):
+        return self.count()
+
     def part2(self):
-        count = 0
-        depths = self.getInputListAsType(int)
-
-        for x in range(3, len(depths)):
-            if depths[x] > depths[x-3]:
-                count += 1
-
-        return count
+        return self.count(3)
 
 
 if __name__ == '__main__':