Post-christmas cleaning.

This commit is contained in:
Mikael CAPELLE
2023-01-06 13:48:18 +01:00
parent dd80b30e26
commit 7533dd0b11
5 changed files with 22 additions and 31 deletions

View File

@@ -2,19 +2,8 @@
import sys
lines = sys.stdin.readlines()
blocks = sys.stdin.read().split("\n\n")
values = sorted(sum(map(int, block.split())) for block in blocks)
# we store the list of calories for each elf in values, and we use the last element
# of values to accumulate
values: list[int] = [0]
for line in lines:
if not line.strip():
values = values + [0]
else:
values[-1] += int(line.strip())
# part 1
print(f"answer 1 is {max(values)}")
# part 2
print(f"answer 2 is {sum(sorted(values)[-3:])}")
print(f"answer 1 is {values[-1]}")
print(f"answer 2 is {sum(values[-3:])}")