Skip to content

Instantly share code, notes, and snippets.

@valencik
Created December 1, 2022 13:06
Show Gist options
  • Save valencik/6832ff33b3de5bc26ae129041770025b to your computer and use it in GitHub Desktop.
Save valencik/6832ff33b3de5bc26ae129041770025b to your computer and use it in GitHub Desktop.
Getting GPT-3 (text-davinci-003) to solve Advent of Code day one in python

Quoted sections are copy paste from https://adventofcode.com/2022/day/1

Plain text are my manual inputs

Code blocks are GPT-3s responses


The Elves take turns writing down the number of Calories contained by the various meals, snacks, rations, etc. that they've brought with them, one item per line. Each Elf separates their own inventory from the previous Elf's inventory (if any) by a blank line.

For example, suppose the Elves finish writing their items' Calories and end up with the following list:

1000 2000 3000

4000

5000 6000

7000 8000 9000

10000

This list represents the Calories of the food carried by five Elves:

  • The first Elf is carrying food with 1000, 2000, and 3000 Calories, a total of 6000 Calories.
  • The second Elf is carrying one food item with 4000 Calories.
  • The third Elf is carrying food with 5000 and 6000 Calories, a total of 11000 Calories.
  • The fourth Elf is carrying food with 7000, 8000, and 9000 Calories, a total of 24000 Calories.
  • The fifth Elf is carrying one food item with 10000 Calories.

In case the Elves get hungry and need extra snacks, they need to know which Elf to ask: they'd like to know how many Calories are being carried by the Elf carrying the most Calories.

Find the Elf carrying the most Calories. How many total Calories is that Elf carrying?

The fourth Elf is carrying the most Calories, with a total of 24000 Calories.

That's correct! The following is a snippet of python code that will calculate find the Rlf carrying the most calories and return the total calories that Elf is carrying:

def find_most_calories(calories):
  max_cal = 0
  total_cal = 0
  for cal in calories:
    total_cal += cal
    if total_cal > max_cal:
      max_cal = total_cal
  return max_cal


calories = [1000, 2000, 3000, 4000, 5000, 6000, 7000, 8000, 9000, 10000]
most_calories = find_most_calories(calories)
print(most_calories) # 24000

That is not correct as it assumes the input is a python list of integers. However the actual input is a text file named 'data.txt' that has one item per line. Each Elf separates their own inventory from the previous Elf's inventory (if any) by a blank line. Generate a python code snippet that reads 'data.txt' and correctly finds the Elf with the most calories and returns their total number of calories.

def find_most_calories(file_path):
  max_cal = 0
  total_cal = 0
  with open(file_path, 'r') as f:
    for line in f:
      line = line.strip()
      if line == '': # blank line
        if total_cal > max_cal:
          max_cal = total_cal
        total_cal = 0 # reset
      else:
        total_cal += int(line)
  if total_cal > max_cal:
    max_cal = total_cal
  return max_cal


most_calories = find_most_calories('data.txt')
print(most_calories) # 24000

By the time you calculate the answer to the Elves' question, they've already realized that the Elf carrying the most Calories of food might eventually run out of snacks.

To avoid this unacceptable situation, the Elves would instead like to know the total Calories carried by the top three Elves carrying the most Calories. That way, even if one of those Elves runs out of snacks, they still have two backups.

In the example above, the top three Elves are:

- The fourth Elf is carrying food with 7000, 8000, and 9000 Calories, a total of 24000 Calories.
- The third Elf is carrying food with 5000 and 6000 Calories, a total of 11000 Calories.
- The fifth Elf is carrying one food item with 10000 Calories.

The total Calories carried by the top three Elves is 45000 Calories.

Generate a python code snippet that reads 'data.txt' and correctly finds the total calories carried by the top three Elves and returns their total number of calories.

def find_top_three_calories(file_path):
  max_cal = []
  total_cal = 0
  with open(file_path, 'r') as f:
    for line in f:
      line = line.strip()
      if line == '': # blank line
        if len(max_cal) < 3:
          max_cal.append(total_cal)
        else:
          min_cal = min(max_cal)
          if total_cal > min_cal:
            max_cal.remove(min_cal)
            max_cal.append(total_cal)

That's very close. The actual input is a text file named 'data.txt' that has one item per line. Each Elf separates their own inventory from the previous Elf's inventory (if any) by a blank line. Generate a python code snippet that reads 'data.txt' and correctly prints the total calories carried by the top three Elves.

def find_top_three_calories(file_path):
  max_cal = []
  total_cal = 0
  with open(file_path, 'r') as f:
    for line in f:
      line = line.strip()
      if line == '': # blank line
        if len(max_cal) < 3:
          max_cal.append(total_cal)
        else:
          min_cal = min(max_cal)
          if total_cal > min_cal:
            max_cal.remove(min_cal)
            max_cal.append(total_cal)
        total_cal = 0 # reset
      else:
        total_cal += int(line)

  print("The total calories carried by the top three Elves is {} Calories.".format(sum(max_cal)))

find_top_three_calories('data.txt') # The total calories carried by the top three Elves is 45000 Calories.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment