2022: d01: ex2: add solution

This commit is contained in:
Bruno BELANYI 2022-12-01 09:57:27 +01:00
parent a064fe7199
commit 6f0ffd34db
1 changed files with 21 additions and 0 deletions

21
2022/d01/ex2/ex2.py Executable file
View File

@ -0,0 +1,21 @@
#!/usr/bin/env python
import sys
def solve(input: list[list[int]]) -> int:
totals = sorted(map(lambda l: sum(l, start=0), input), reverse=True)
return sum(totals[:3])
def main() -> None:
input = [
[int(line) for line in group.splitlines()]
for group in sys.stdin.read().split("\n\n")
if group
]
print(solve(input))
if __name__ == "__main__":
main()