01_fuel.py 348 B

123456789101112131415161718
  1. #!/usr/bin/env python3
  2. import fileinput
  3. masses = [int(line) for line in fileinput.input()]
  4. def fuel(mass):
  5. return int(mass / 3) - 2
  6. def fuelrec(mass):
  7. f = fuel(mass)
  8. total = 0
  9. while f >= 0:
  10. total += f
  11. mass = f
  12. f = fuel(mass)
  13. return total
  14. print(sum(map(fuel, masses)))
  15. print(sum(map(fuelrec, masses)))