From 32fe21038a75a3f6da911c92247e422885cb521b Mon Sep 17 00:00:00 2001 From: Thomas Voss Date: Wed, 1 Dec 2021 08:55:41 +0100 Subject: Make use of the magic zip function --- 2021/01/puzzles.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to '2021') 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() -- cgit v1.2.3