update for part 2

clement7903 · clement7903 · commit f8d978727a0a · 2022-12-01T14:56:24.000+08:00
diff --git a/1.caloriecounting.py b/1.caloriecounting.py
@@ -13,16 +13,14 @@
       total = 0
     
     elif line == 'STOP':
+      top3 = 0
       for _ in range(3):
         highest = heappop(lst)
-        print(-highest[0])
+        top3 += -highest[0]
+      print(top3)
     
     else:
       total += int(line)
     
   except EOFError:
-    break
-
-  
-
-
+    break
