DEV Community


Advent Of Code 2019: Day 1 solutions in Python

(they/them). Technologist + designer. Interested in ethics, accessibility, inclusivity and AI/ML. Often having fun with css.
Originally published at ・Updated on ・3 min read

Originally posted on

I've been really excited to work on the advent of code challenges this year again. If you don't know the concept, it's basically a, advent calendar with programming puzzles that gradually get more difficult. Each day has two challenges that are related to each other, and they can be solved using any coding language or software.

While I'm not planning to compete for points in any leaderboard, my goal for this year is to solve as many of the puzzles as possible and explain my solutions for some of the fun ones on here. I'll mainly be using Python.

Puzzle day 1

In today's puzzle, we get one input file which contains the mass of modules on a space ship. Based on this input, we are supposed to find the total fuel needed to launch the ship based on a formula.

πŸ•΅οΈβ€β™€οΈ Read the full problem.

Part 1

I solved part 1 in one two lines of code:

input = np.array([input])
output = np.sum(np.floor(input / 3) - 2)

First, we create a numpy array from the input, so we can avoid having to manually loop through each row of the input.

Then, we divide the array by 3, floored (rounded down) the result, and substracted 2.

πŸ‘‰ View my solution on GitHub.

Part 2

We found the amount of fuel needed to accommodate for the weight of the modules. However, the fuel also adds weight to the space ship, meaning that we need to add extra fuel to our ship, using the same formula, to make up for the extra weight the fuel added 🀯.

In short, for each module we'll have to keep repearting the same formula until we reach 0. Then, we just need to take the sum and we know the total amount of fuel.

First, we create an array from the input file that the advent of code website provided. Then, we create a new value for the total fuel, which equals zero for now.

input_array = [...]
total_fuel = 0

This time I did decide to go for a for-loop, although I'm looking for a way of simplifying it.

For each module, we want to calculate the amount of fuel needed (and the amount of fuel needed for the fuel, and the amount of fuel needed for that fuel, and so on) until we reach 0. We can solve this part with a while-loop.

for module in input_array:
    while True:
        // calculate the amount of fuel needed
        if new_fuel > 0:
            // add the amount of fuel total_fuel

Now the calculation part. This will be similar to the first part, since we're using the same formula:

new_fuel = np.floor(new_fuel / 3) - 2

Then the only thing left to do is adding the new_fuel to the total_fuel if it's a positive number.

for module in input_array:
    new_fuel = mod
    while True:
        new_fuel = np.floor(new_fuel / 3) - 2
        if new_fuel > 0:
            total_fuel += new_fuel

Outside of the for-loop we can now print the total_fuel, which is the output for part 2 of the exercise.

πŸ‘‰ View my solution on GitHub.

I will try to publish and explain as many of my solutions as possible over the next couple of weeks, if time allows it. You can also keep an eye on my GitHub account for updates.

Discussion (2)

jaycohh profile image

I made an account here just to say thank you for taking the time to write this up! I was able to solve the first part of the problem, and knew to use a for while loop in the second but didn't know how to apply it!

fossheim profile image
Sarah Author

Glad I could help! πŸ€—

Forem Open with the Forem app