aoc2021/day01.py

35 lines
652 B
Python

from tools.aoc import AOCDay
class Day(AOCDay):
inputs = [
[
(7, "test_input01"),
(1602, "input01")
],
[
(5, "test_input01"),
(1633, "input01")
]
]
def count(self, step: int = 1) -> int:
count = 0
depths = self.getInput(int)
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):
return self.count(3)
if __name__ == '__main__':
day = Day(2021, 1)
day.run(verbose=True)