diff --git a/src/main/kotlin/adventofcode/year2024/Day22MonkeyMarket.kt b/src/main/kotlin/adventofcode/year2024/Day22MonkeyMarket.kt index 5e553b5..cb7febd 100644 --- 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 }