about summary refs log tree commit diff
path: root/users/wpcarro/scratch/advent-of-code-2019/day_1.py
blob: bd4024e3ec7d65519ac6feff6b9653bdbbb6146e (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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from math import floor

xs = [
    102473,
    84495,
    98490,
    68860,
    62204,
    72810,
    65185,
    145951,
    77892,
    108861,
    70764,
    67286,
    74002,
    80773,
    52442,
    131505,
    107162,
    126993,
    59784,
    64231,
    91564,
    68585,
    98735,
    69020,
    77332,
    60445,
    65826,
    111506,
    95431,
    146687,
    135119,
    86804,
    95915,
    85434,
    111303,
    148127,
    132921,
    136213,
    89004,
    143137,
    144853,
    143017,
    104386,
    100612,
    54760,
    63813,
    144191,
    84481,
    69718,
    84936,
    98621,
    124993,
    92736,
    60369,
    137284,
    101902,
    112726,
    51784,
    126496,
    85005,
    101661,
    137278,
    136637,
    90340,
    100209,
    53683,
    50222,
    132060,
    98797,
    139054,
    135638,
    100632,
    137849,
    125333,
    103981,
    76954,
    134352,
    74229,
    93402,
    62552,
    50286,
    57066,
    98439,
    120708,
    117827,
    107884,
    72837,
    148663,
    125645,
    61460,
    120555,
    142473,
    106668,
    58612,
    58576,
    143366,
    90058,
    121087,
    89546,
    126161,
]


def fuel_for_mass(x):
    """Return the amount of fuel (in mass) required for a mass of X. The total
    amount of fuel includes the amount of fuel required for the fuel itself,
    since fuel also has a mass weights."""
    mass_fuel = floor(x / 3) - 2
    if mass_fuel < 0:
        return 0
    else:
        fuel_fuel = fuel_for_mass(mass_fuel)
        return mass_fuel + fuel_fuel


print(sum(fuel_for_mass(x) for x in xs))