refactor: shorter size.contains

scarf005 · Dec 8, 2024 · 6431f13 · 6431f13
1 parent 62968c8
commit 6431f13
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/2024/day08.scala b/2024/day08.scala
@@ -12,10 +12,9 @@ case class Context(size: Size, antennaes: Map[Char, Vector[Pos]]):
     antennaes.values.flatMap(_.combinationsN(2).flatMap(fn)).toSet
 
   lazy val part1 =
-    solve((a, b) => Vector(a + (a - b), b + (b - a))).filter(size.contains).size
+    solve((a, b) => Vector(a + (a - b), b + (b - a))).filter(size(_)).size
   lazy val part2 = solve((a, b) =>
-    (a deltas b).takeWhile(size.contains)
-      ++ (b deltas a).takeWhile(size.contains),
+    (a deltas b).takeWhile(size(_)) ++ (b deltas a).takeWhile(size(_)),
   ).size
 
 object Context: