2022 day 1

2022/day1/day1.py

@@ -0,0 +1,14 @@
+def main(content):
+   inventories = [x.rstrip().split("\n") for x in content.split("\n\n")]
+   calories = sorted((sum(int(y) for y in x) for x in inventories), reverse=True)
+   print("Part 1: ", calories[0])
+   print("Part 2: ", sum(calories[:3]))
+
+
+if __name__ == "__main__":
+    import sys
+    infile = sys.argv[1]
+    with open(infile) as f:
+        content = f.read()
+        main(content)
+