diff --git a/2024/d02/ex1/ex1.py b/2024/d02/ex1/ex1.py new file mode 100755 index 0000000..1c1bc86 --- /dev/null +++ b/2024/d02/ex1/ex1.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python + +import itertools +import sys + +Report = list[int] + + +def solve(input: list[str]) -> int: + def parse(input: list[str]) -> list[Report]: + return [[int(n) for n in line.split()] for line in input] + + def is_safe(report: Report) -> bool: + def is_increasing_safe(report: Report) -> bool: + return all(1 <= (b - a) <= 3 for a, b in itertools.pairwise(report)) + + return is_increasing_safe(report) or is_increasing_safe(report[::-1]) + + return sum(is_safe(report) for report in parse(input)) + + +def main() -> None: + input = sys.stdin.read().splitlines() + print(solve(input)) + + +if __name__ == "__main__": + main()