def load(filename: str) -> list[int]:
	elves = []
	with open(filename) as f:
		calories = 0
		for l in f:
			cals = l.strip()
			if cals == "":
				elves.append(calories)
				calories = 0
				continue

			calories += int(cals)

		elves.append(calories)

		return elves

demo = sorted(load("demo.txt"), reverse=True)
data = sorted(load("data.txt"), reverse=True)

# Part 1
assert max(demo) == 24000
print("Part 1: ", max(data))

# Part 2
assert sum(demo[0:3]) == 45000
print("Part 2: ", sum(data[0:3]))