Make use of the magic zip function

1 files changed, 1 insertions, 4 deletions

diff --git a/2021/01/puzzles.py b/2021/01/puzzles.py
index 0a894ca..d9a704b 100644
--- a/2021/01/puzzles.py
+++ b/2021/01/puzzles.py
@@ -4,10 +4,7 @@ def main() -> None:
 	with open("input", "r") as f:
 		nums = [int(n) for n in f.readlines()]
 
-	# START PART 2
-	nums = [nums[i] + nums[i - 1] + nums[i - 2] for i in range(2, len(nums))]
-	# END PART 2
-	print(sum(nums[i] > nums[i - 1] for i in range(1, len(nums))))
+	print(sum(n < m for n, m in zip(nums, nums[DELTA:])))
 
 if __name__ == "__main__":
 	main()