Yet, minor cleanup

1 files changed, 5 insertions, 10 deletions

diff --git a/2024/05/solve.py b/2024/05/solve.py
index b17c986..31b75c0 100755
--- a/2024/05/solve.py
+++ b/2024/05/solve.py
@@ -7,19 +7,14 @@ numss = [[int(c) for c in l.split(",")] for l in L[1].strip().split("\n")]
 
 
 def check(nums):
-    pos = (-1, -1)
-    ordered = True
     for i, a in enumerate(nums):
         for j, b in enumerate(nums[i:]):
             if [b, a] in ordering:
-                ordered = False
-                pos = (i, i + j)
-                break
-        if not ordered:
-            break
-    return (pos, ordered)
+                return ((i, i + j), False)
+    return (None, True)
 
 
+# "bubble sort"
 def fix(nums):
     while True:
         pos, ordered = check(nums)
@@ -32,12 +27,12 @@ def solve():
     res1 = 0
     res2 = 0
 
-    for n, nums in enumerate(numss):
+    for nums in numss:
         _, ordered = check(nums)
         if ordered:
             res1 += nums[len(nums) // 2]
         else:
-            fixed = fix(nums.copy())
+            fixed = fix(nums)
             res2 += fixed[len(fixed) // 2]
 
     print(res1)