02_dive.py 365 B

123456789101112131415161718
  1. #!/usr/bin/env python3
  2. import sys
  3. DELTA = {'forward': (1, 0), 'up': (0, -1), 'down': (0, 1)}
  4. def parse(line):
  5. direction, amount = line.split()
  6. return DELTA[direction], int(amount)
  7. x = y = aimed_y = 0
  8. for (dx, dy), amount in map(parse, sys.stdin):
  9. x += dx * amount
  10. y += dy * amount
  11. aimed_y += dx * y * amount
  12. print(x * y)
  13. print(x * aimed_y)