mirror of
https://github.com/Xevion/advent-of-code.git
synced 2025-12-06 05:14:22 -06:00
19 lines
532 B
Python
19 lines
532 B
Python
import os
|
|
import sys
|
|
import math
|
|
|
|
PATH = os.path.join(sys.path[0], '..', 'input')
|
|
DATA = list(map(int, open(PATH, 'r').readlines()))
|
|
|
|
BASIC_FUEL = lambda mass : math.floor(mass / 3) - 2
|
|
ADVANCED_FUEL = lambda mass : BASIC_FUEL(mass) + (BASIC_FUEL(BASIC_FUEL(mass)) if BASIC_FUEL(mass) > 0 else 0)
|
|
|
|
# def ADVANCED_FUEL(mass):
|
|
# total = 0
|
|
# while mass > 0:
|
|
# mass = BASIC_FUEL(mass)
|
|
# total += mass if mass >= 0 else 0
|
|
# return total
|
|
|
|
print(sum(map(BASIC_FUEL, DATA)))
|
|
print(sum(map(ADVANCED_FUEL, DATA))) |