[2024/22] Refactor part 2

pfolta · Dec 23, 2024 · c842854 · c842854
1 parent dab3b6a
commit c842854
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/main/kotlin/adventofcode/year2024/Day22MonkeyMarket.kt b/src/main/kotlin/adventofcode/year2024/Day22MonkeyMarket.kt
@@ -11,12 +11,17 @@ class Day22MonkeyMarket(customInput: PuzzleInput? = null) : Puzzle(customInput)
     override fun partTwo() =
         buildMap {
             secretNumbers
-                .map { secretNumber -> secretNumber.evolve(2000).map { i -> i % 10 }.toList() }
+                .map { secretNumber ->
+                    secretNumber
+                        .evolve(2000)
+                        .map { number -> number % 10 }
+                        .toList()
+                }
                 .forEach { sequence ->
                     sequence
                         .windowed(5, 1)
-                        .map { it.zipWithNext { a, b -> b - a } to it.last() }
-                        .distinctBy { (a, _) -> a }
+                        .map { slice -> slice.zipWithNext { a, b -> b - a } to slice.last() }
+                        .distinctBy { (changes, _) -> changes }
                         .forEach { (key, value) ->
                             this[key] = (this[key] ?: 0L) + value
                         }