2022-python/output/day_01.py

@@ -0,0 +1,26 @@
+from output import ints
+
+
+def solve(data):
+    E = data.split("\n\n")
+    p1 = 0
+    C = []
+    for e in E:
+        e = sum(ints(e))
+        p1 = max(p1, e)
+        C.append(e)
+    p2 = sum(sorted(C, reverse=True)[:3])
+    return p1, p2
+
+
+if __name__ == "__main__":
+    with open("./input/01.txt", "r") as f:
+        inp = f.read().strip()
+
+    p1, p2 = solve(inp)
+
+    print(p1)
+    print(p2)
+
+    assert p1 == 66616
+    assert p2 == 199172