From f3a3e1fca3f44b62e4753c284f10fd681d272dcd Mon Sep 17 00:00:00 2001
From: Bert Peters <bert@bertptrs.nl>
Date: Wed, 25 Dec 2024 10:34:40 +0100
Subject: [PATCH] More numpy for speed

---
 2024/src/aoc/days/day25.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/2024/src/aoc/days/day25.py b/2024/src/aoc/days/day25.py
index a500a3e..62ed05a 100644
--- 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