07_crabs.py 333 B

12345678910
  1. #!/usr/bin/env python3
  2. import sys
  3. def fuels(crabs, fuel):
  4. for dest in range(min(crabs), max(crabs) + 1):
  5. yield sum(fuel(abs(dest - crab)) for crab in crabs)
  6. crabs = list(map(int, sys.stdin.readline().split(',')))
  7. print(min(fuels(crabs, lambda dist: dist)))
  8. print(min(fuels(crabs, lambda dist: dist * (dist + 1) // 2)))