01.py

#!/usr/bin/env python3

import math

input = """116115
58728
102094
104856
86377
97920
101639
95328
103730
57027
83080
57748
101606
54629
90901
59983
109795
123270
141948
92969
149805
143555
141387
136357
90236
63577
127108
130012
88223
51426
117663
63924
56251
108505
89625
126994
120237
99351
136948
123702
129849
93541
110900
63759
58537
132943
118213
104274
84606
125256
76355
116711
79344
66355
117654
116026
80244
129786
73054
119806
90941
53877
96707
58226
101666
53819
54558
77342
149653
87843
54388
128862
55752
89962
147224
118486
56910
124854
57052
55495
62530
128104
68788
60915
62155
123614
115522
116920
101263
92339
92234
81542
78062
137207
92082
120032
136537
109035
115819
75955""".split("\n")



# Teil 1

def fuel(mass):
    return math.floor(mass/3) - 2

total_fuel = 0
for mass in input:
    total_fuel = total_fuel + fuel(int(mass))

print(total_fuel)



# Teil 2
def fuel_total(mass):
    total_fuel = fuel(mass)
    
    additional_mass = total_fuel
    while True:
        additional_fuel = fuel(additional_mass)
        if additional_fuel > 0:
            total_fuel = total_fuel + additional_fuel
            additional_mass = additional_fuel
        else:
            break
    
    return total_fuel


fuel_including_fuel = 0
for mass in input:
    fuel_including_fuel = fuel_including_fuel + fuel_total(int(mass))

print(fuel_including_fuel)