21 lines
533 B
Python
21 lines
533 B
Python
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 = "d1.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()
|