aoc2022_d01.py

aoc2022_d01.py

input = """
INPUT HERE
"""

totals = {}
index = 0

amounts = input.split("\n\n")
for elf in amounts:
    calories = elf.split("\n")
    if '' in calories:
        calories.remove('')

    totals[index] = 0
    for calorie in calories:
        totals[index] = totals[index] + int(calorie)
    
    index = index + 1

in_order = sorted(totals.values(), reverse=True)
print("Part 1:")
print(in_order[0])

print("Part 2:")
print(in_order[0] + in_order[1] + in_order[2])