Improvements

1 files changed, 8 insertions, 13 deletions

diff --git a/2022/01/solve.py b/2022/01/solve.py
index de9aea8..7ad0cf7 100644
--- a/2022/01/solve.py
+++ b/2022/01/solve.py
@@ -1,27 +1,22 @@
-data = [block.split("\n") for block in open("input").read().split("\n\n")]
+data = [[int(calory) for calory in block.split("\n") if calory != ''] for block in open("input").read().split("\n\n")]
 
 # i don't know list comprehension
 
 def part1():
     res = 0
     for block in data:
-        sum = 0
-        for calory in block:
-            if len(calory) > 0:
-                sum += int(calory)
-        if sum > res:
-            res = sum
+        s = sum(block)
+        if s > res:
+            res = s
     return res
 
 def part2():
     top3 = [0,0,0]
     for block in data:
-        s = 0
-        for calory in block:
-            if len(calory) > 0:
-                s += int(calory)
-        if s > min(top3):
-            top3.remove(min(top3))
+        s = sum(block)
+        m = min(top3)
+        if s > m:
+            top3.remove(m)
             top3.append(s)
     return sum(top3)