blob: d1c57c88621c5ee08b21683f33c62d79092a34e0 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
import math
import sys
def get_fuel(mass):
fuel = math.floor(mass / 3) - 2
return 0 if fuel <= 0 else fuel + get_fuel(fuel)
def pt1(_in):
s = 0
for line in _in:
mass = int(line)
if mass == 0:
break
fuel = math.floor(mass / 3) - 2
s += fuel
return s
def pt2(input):
return sum([get_fuel(int(line)) for line in input])
if __name__ == "__main__":
import cProfile
input = open("../input/01", "r").readlines()
cProfile.run("pt1(input)")
cProfile.run("pt2(input)")
print(pt1(input))
print(pt2(input))
|
