aoc2022/d1.py

from lib import *

def solve(input: Input, second=False):
    ps = input.text.split("\n\n")
    xss = map(lambda p: map(int, p.splitlines()), ps)
    xs = list(map(sum, xss))
    if not second:
        return max(xs)
    xs.sort()
    return sum(xs[-3:])

def main():
    DAY_INPUT = "i1.txt"
    print("Solution 1:", solve(Input(DAY_INPUT)))
    assert solve(Input(DAY_INPUT)) == 69883
    print("Solution 2:", solve(Input(DAY_INPUT), True))
    assert solve(Input(DAY_INPUT), True) == 207576

if __name__ == "__main__":
    main()