From 7dc8aaa7f8e17a74c0b0c59690ac9737395452d8 Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Thu, 1 Dec 2022 06:30:54 +0100
Subject: Improvements

---
 2022/01/solve.py | 21 ++++++++-------------
 1 file changed, 8 insertions(+), 13 deletions(-)

(limited to '2022/01/solve.py')

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)
 
-- 
cgit v1.2.3