05_jumps.py 382 B

1234567891011121314151617
  1. #!/usr/bin/env python3
  2. import sys
  3. def escape(maze, decrease):
  4. vis = [0] * len(maze)
  5. steps = 0
  6. pos = 0
  7. while 0 <= pos < len(maze):
  8. jump = maze[pos] + vis[pos]
  9. vis[pos] += -1 if decrease and jump >= 3 else 1
  10. pos += jump
  11. steps += 1
  12. return steps
  13. maze = list(map(int, sys.stdin))
  14. print(escape(maze, False))
  15. print(escape(maze, True))