18_traps.py 426 B

1234567891011121314
  1. #!/usr/bin/env python3
  2. import sys
  3. def count_safe(first, nrows):
  4. row = [False] + [c == '^' for c in first] + [False]
  5. safe = len(row) - sum(row) - 2
  6. for step in range(nrows - 1):
  7. row[1:-1] = (row[i - 1] != row[i + 1] for i in range(1, len(row) - 1))
  8. safe += len(row) - sum(row) - 2
  9. return safe
  10. first = sys.stdin.readline().rstrip()
  11. print(count_safe(first, 40))
  12. print(count_safe(first, 400000))