40 lines
1,009 B
Python
40 lines
1,009 B
Python
|
#!/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):
|
||
|
|
for a, b in itertools.pairwise(report):
|
||
|
|
if not 1 <= (b - a) <= 3:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
return is_increasing_safe(report) or is_increasing_safe(report[::-1])
|
||
|
|
|
||
|
|
# Dampening by brute force
|
||
|
|
def is_dampened_safe(report: Report) -> bool:
|
||
|
|
for i in range(len(report)):
|
||
|
|
dampened_report = report[:i] + report[i + 1 :]
|
||
|
|
if is_safe(dampened_report):
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
return sum(is_dampened_safe(report) for report in parse(input))
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
input = sys.stdin.read().splitlines()
|
||
|
|
print(solve(input))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|