01_sonar.py 361 B

12345678910111213
  1. #!/usr/bin/env python3
  2. import sys
  3. def slide(depths, winsize):
  4. return [sum(depths[i:i + winsize])
  5. for i in range(len(depths) - winsize + 1)]
  6. def increased(depths):
  7. return sum(depths[i] < depths[i + 1] for i in range(len(depths) - 1))
  8. depths = list(map(int, sys.stdin))
  9. print(increased(slide(depths, 1)))
  10. print(increased(slide(depths, 3)))