More numpy for speed

2025-12-25 12:50:32 +01:00 · 2024-12-25 10:34:40 +01:00
parent 4a479c1646
commit f3a3e1fca3
1 changed files with 4 additions and 5 deletions
--- a/2024/src/aoc/days/day25.py
+++ b/2024/src/aoc/days/day25.py
@@ -20,11 +20,10 @@ class DayRunner(CombinedRunner):
            else:
                keys.append(heights)

-        fitting = 0
+        locks = numpy.stack(locks, axis=0)

-        for key in keys:
-            for lock in locks:
-                if numpy.all((key + lock) <= 7):
-                    fitting += 1
+        fitting = sum(
+            numpy.count_nonzero(numpy.all((key + locks) <= 7, axis=1)) for key in keys
+        )

        return fitting, None