Day 01 in python

This commit is contained in:
aru 2021-12-01 12:10:12 +01:00
parent 0da732397b
commit aaccc3c608
2 changed files with 23 additions and 0 deletions

10
01/example_input Normal file
View File

@ -0,0 +1,10 @@
199
200
208
210
200
207
240
269
260
263

13
01/solution.py Normal file
View File

@ -0,0 +1,13 @@
import sys
with open(sys.argv[1]) as f:
numbers = list(map(int, f.readlines()))
def part1(numbers):
return sum(1 for (cur, prev) in zip(numbers[1:], numbers) if cur > prev)
def part2(numbers):
return part1([sum(x) for x in zip(numbers, numbers[1:], numbers[2:])])
print(part1(numbers))
print(part2(numbers))