Post-christmas cleaning.
This commit is contained in:
19
2022/day1.py
19
2022/day1.py
@@ -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:])}")
|
||||
|
Reference in New Issue
Block a user