diff --git a/2022/d01/ex2/ex2.py b/2022/d01/ex2/ex2.py new file mode 100755 index 0000000..64da90e --- /dev/null +++ b/2022/d01/ex2/ex2.py @@ -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()