03_trees.py 448 B

12345678910111213
  1. #!/usr/bin/env python3
  2. import sys
  3. from functools import reduce
  4. from operator import mul
  5. def count(treemap, right, down):
  6. return sum(line[((i + 1) * right) % len(line)]
  7. for i, line in enumerate(treemap[down::down]))
  8. treemap = [[x == '#' for x in line.rstrip()] for line in sys.stdin]
  9. print(count(treemap, 3, 1))
  10. print(reduce(mul, (count(treemap, r, d)
  11. for r, d, in ((1, 1), (3, 1), (5, 1), (7, 1), (1, 2)))))